Update README.md
Browse files
README.md
CHANGED
@@ -44,7 +44,7 @@ Then you can use `FluxPipeline` to run the model
|
|
44 |
import torch
|
45 |
from diffusers import FluxPipeline
|
46 |
|
47 |
-
pipe = FluxPipeline.from_pretrained("
|
48 |
pipe.enable_model_cpu_offload() #save some VRAM by offloading the model to CPU. Remove this if you have enough GPU power
|
49 |
|
50 |
prompt = "A cat holding a sign that says hello world"
|
|
|
44 |
import torch
|
45 |
from diffusers import FluxPipeline
|
46 |
|
47 |
+
pipe = FluxPipeline.from_pretrained("FILM6912/FLUX.1-schnell", torch_dtype=torch.bfloat16)
|
48 |
pipe.enable_model_cpu_offload() #save some VRAM by offloading the model to CPU. Remove this if you have enough GPU power
|
49 |
|
50 |
prompt = "A cat holding a sign that says hello world"
|