Hugging Face's logo

Spaces:
OpenSound
/
SoloAudio
Running on Zero

App Files Community
OpenSound commited on
Commit
a748bcc
·
verified ·
1 Parent(s): 7f7f21e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -40,7 +40,7 @@ clapmodel.to(device)
40
 
41
  autoencoder = Autoencoder(autoencoder_path, 'stable_vae', quantization_first=True)
42
  autoencoder.eval()
43
- autoencoder.to(device)
44
  unet = UDiT(
45
  **diff_config['diffwrap']['UDiT']
46
  ).to(device)
 
40
 
41
  autoencoder = Autoencoder(autoencoder_path, 'stable_vae', quantization_first=True)
42
  autoencoder.eval()
43
+ autoencoder = autoencoder.float().to(device)
44
  unet = UDiT(
45
  **diff_config['diffwrap']['UDiT']
46
  ).to(device)