How to use JuliaML/Cosmos3-Super-Text2Image-4Step-INT4-G64-BF16 with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("JuliaML/Cosmos3-Super-Text2Image-4Step-INT4-G64-BF16", dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0]
How to use JuliaML/Cosmos3-Super-Text2Image-4Step-INT4-G64-BF16 with MLX:
# Download the model from the Hub pip install huggingface_hub[hf_xet] huggingface-cli download --local-dir Cosmos3-Super-Text2Image-4Step-INT4-G64-BF16 JuliaML/Cosmos3-Super-Text2Image-4Step-INT4-G64-BF16
