Hugging Face's logo

Spaces:
Yhhxhfh
/
Vvvvv
Runtime error

App Files Community
Yhhxhfh commited on
Commit
a495bf6
·
verified ·
1 Parent(s): af96a46

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -229,7 +229,7 @@ unified_model.to(torch.device("cpu"))
229
  musicgen_tokenizer = global_tokenizers["musicgen-small"]
230
  musicgen_model = global_models["musicgen-small"]
231
  stable_diffusion_pipeline = global_models["stable-diffusion-v1-4"]
232
- stable_diffusion_pipeline.enable_model_cpu_offload()
233
  img2img_pipeline = global_models["stable-diffusion-2-1-base"]
234
  img2img_pipeline.enable_model_cpu_offload()
235
 
 
229
  musicgen_tokenizer = global_tokenizers["musicgen-small"]
230
  musicgen_model = global_models["musicgen-small"]
231
  stable_diffusion_pipeline = global_models["stable-diffusion-v1-4"]
232
+ #stable_diffusion_pipeline.enable_model_cpu_offload()
233
  img2img_pipeline = global_models["stable-diffusion-2-1-base"]
234
  img2img_pipeline.enable_model_cpu_offload()
235