mingyi456/

shuttle-jaguar-DF11

How to Use

diffusers

1. Install the DFloat11 pip package (installs the CUDA kernel automatically; requires a CUDA-compatible GPU and PyTorch installed):
   bash

   1pip install dfloat11[cuda12] 2# or if you have CUDA version 11: 3# pip install dfloat11[cuda11]
2. To use the DFloat11 model, run the following example code in Python:
   python

   1import torch 2from diffusers import FluxPipeline 3from dfloat11 import DFloat11Model 4 5pipe = FluxPipeline.from_pretrained("shuttleai/shuttle-jaguar", torch_dtype=torch.bfloat16) 6pipe.enable_model_cpu_offload() 7 8DFloat11Model.from_pretrained('mingyi456/shuttle-jaguar-DF11', device='cpu', bfloat16_model=pipe.transformer) 9 10prompt = "A futuristic cityscape at sunset, with flying cars, neon lights, and reflective water canals" 11image = pipe( 12 prompt, 13 guidance_scale=0.0, 14 num_inference_steps=4, 15 max_sequence_length=256, 16 generator=torch.Generator("cpu").manual_seed(0) 17).images[0] 18image.save("shuttle-jaguar.png")

ComfyUI