Instructions to use hf-internal-testing/tiny-cosmos3-modular-pipe with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use hf-internal-testing/tiny-cosmos3-modular-pipe with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("hf-internal-testing/tiny-cosmos3-modular-pipe", dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0] - Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- Draw Things
- DiffusionBee
Shrink tiny Cosmos3 transformer for CPU tests
#2
by atharvajoshi10 - opened
Reduce the dummy transformer to one layer, one attention head, and hidden size 6 while preserving the full tokenizer vocabulary and the vision, sound, and action paths.
YiYiXu changed pull request status to merged