How to use nvidia/Qwen-Image-Flash with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("nvidia/Qwen-Image-Flash", dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0]
Hi! Thank you for your excellent work.
I was wondering if you have any plans to release the training pipeline.Also, could you share approximately how many GPUs were used for training and roughly how long the training took?
· Sign up or log in to comment
