Ely-testa commited on
Commit
b18e629
·
verified ·
1 Parent(s): e781192
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ import torch
6
  from diffusers import FluxPipeline
7
 
8
  pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
9
- pipe.enable_model_cpu_offload() #save some VRAM by offloading the model to CPU. Remove this if you have enough GPU power
10
 
11
  prompt = "A cat holding a sign that says hello world"
12
  image = pipe(
 
6
  from diffusers import FluxPipeline
7
 
8
  pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
9
+ #pipe.enable_model_cpu_offload() #save some VRAM by offloading the model to CPU. Remove this if you have enough GPU power
10
 
11
  prompt = "A cat holding a sign that says hello world"
12
  image = pipe(