Update README.md

README.md

@@ -19,7 +19,7 @@ The Flux model with 4bit transformer and T5 encoder.
 ```python
 from diffusers import FluxPipeline
 import torch
-pipeline = FluxPipeline("eramth/flux-4bit",torch_dtype=torch.float16)
+pipeline = FluxPipeline("eramth/flux-4bit",torch_dtype=torch.float16).to("cuda")
 image = pipeline(prompt="a cute cat",num_inference_steps=25,guidance_scale=3.5).images[0]
 image