Update README.md

README.md

@@ -23,16 +23,17 @@ import torch
 from diffusers import FluxPipeline
 
 pipe = FluxPipeline.from_pretrained(
-    "Silan10/flux_quantized_half"
+    "Silan10/flux_quantized_half",
+    torch_dtype=torch.float16
 )
 pipe.to("cuda")  # or pipe.enable_model_cpu_offload() for low VRAM
 
-prompt = "A detailed concept art of a futuristic city at sunset"
+prompt = "A detailed concept art of a futuristic cat"
 image = pipe(
     prompt,
     guidance_scale=3.5,
     num_inference_steps=20,
-    max_sequence_length=256,
+    generator=torch.Generator("cpu").manual_seed(0)
 ).images[0]
 
 image.save("flux_half_sample.png")