FILM6912 commited on
Commit
4a027ed
1 Parent(s): d54411a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -52,7 +52,7 @@ Then you can use `FluxPipeline` to run the model
52
  import torch
53
  from diffusers import FluxPipeline
54
 
55
- pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
56
  pipe.enable_model_cpu_offload() #save some VRAM by offloading the model to CPU. Remove this if you have enough GPU power
57
 
58
  prompt = "A cat holding a sign that says hello world"
 
52
  import torch
53
  from diffusers import FluxPipeline
54
 
55
+ pipe = FluxPipeline.from_pretrained("FILM6912/FLUX.1-dev", torch_dtype=torch.bfloat16)
56
  pipe.enable_model_cpu_offload() #save some VRAM by offloading the model to CPU. Remove this if you have enough GPU power
57
 
58
  prompt = "A cat holding a sign that says hello world"