Hugging Face's logo

Spaces:
concauu
/
image_generator
Runtime error

App Files Community
concauu commited on
Commit
233e97e
·
verified ·
1 Parent(s): a511dd1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ login(token=decrypted_token)
27
 
28
 
29
  pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.float32)
30
- # pipe.enable_model_cpu_offload() #save some VRAM by offloading the model to CPU. Remove this if you have enough GPU power
31
 
32
 
33
  def enhance_prompt(user_prompt):
 
27
 
28
 
29
  pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.float32)
30
+ pipe.enable_model_cpu_offload() #save some VRAM by offloading the model to CPU. Remove this if you have enough GPU power
31
 
32
 
33
  def enhance_prompt(user_prompt):