FILM6912
/

FLUX.1-schnell

image-generation

Inference Endpoints

Model card Files Files and versions Community

FILM6912 commited on Sep 8

Commit

ace15bf

•

1 Parent(s): 8174d25

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -44,7 +44,7 @@ Then you can use `FluxPipeline` to run the model
 import torch
 from diffusers import FluxPipeline
-pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16)
 pipe.enable_model_cpu_offload() #save some VRAM by offloading the model to CPU. Remove this if you have enough GPU power
 prompt = "A cat holding a sign that says hello world"

 import torch
 from diffusers import FluxPipeline
+pipe = FluxPipeline.from_pretrained("FILM6912/FLUX.1-schnell", torch_dtype=torch.bfloat16)
 pipe.enable_model_cpu_offload() #save some VRAM by offloading the model to CPU. Remove this if you have enough GPU power
 prompt = "A cat holding a sign that says hello world"