Hugging Face's logo

Spaces:
declare-lab
/
mustango
Runtime error

App Files Community
40
soujanyaporia commited on
Commit
ebd2382
·
verified ·
1 Parent(s): 3dd4ae8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -225,7 +225,10 @@ class Mustango:
225
 
226
 
227
  # Initialize Mustango
228
- mustango = Mustango(device="cpu")
 
 
 
229
  # if torch.cuda.is_available():
230
  # mustango = Mustango()
231
  # else:
 
225
 
226
 
227
  # Initialize Mustango
228
+ mustango = Mustango(device="CPU")
229
+ mustango.vae.to(device_type)
230
+ mustango.stft.to(device_type)
231
+ mustango.model.to(device_type)
232
  # if torch.cuda.is_available():
233
  # mustango = Mustango()
234
  # else: