Hugging Face's logo

Spaces:
cloudfom
/
ao
Sleeping

App Files Community
fomext commited on
Commit
2d15ecf
·
verified ·
1 Parent(s): 8442b37

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -37,7 +37,8 @@ def load_pipeline():
37
  pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-audio-open-1.0",torch_dtype=DTYPE,trust_remote_code=True,revision="main")
38
 
39
  pipe.enable_attention_slicing()
40
- pipe.enable_model_cpu_offload()
 
41
 
42
  print("Pipeline loaded on", DEVICE)
43
 
 
37
  pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-audio-open-1.0",torch_dtype=DTYPE,trust_remote_code=True,revision="main")
38
 
39
  pipe.enable_attention_slicing()
40
+ pipe.to(DEVICE)
41
+
42
 
43
  print("Pipeline loaded on", DEVICE)
44