Update app.py
Browse files
app.py
CHANGED
@@ -22,7 +22,7 @@ with open(diffusion_config, 'r') as fp:
|
|
22 |
|
23 |
v_prediction = diff_config["ddim"]["v_prediction"]
|
24 |
|
25 |
-
clapmodel = ClapModel.from_pretrained("laion/larger_clap_general"
|
26 |
processor = AutoProcessor.from_pretrained('laion/larger_clap_general')
|
27 |
autoencoder = Autoencoder(autoencoder_path, 'stable_vae', quantization_first=True)
|
28 |
autoencoder.eval()
|
|
|
22 |
|
23 |
v_prediction = diff_config["ddim"]["v_prediction"]
|
24 |
|
25 |
+
clapmodel = ClapModel.from_pretrained("laion/larger_clap_general").to(device)
|
26 |
processor = AutoProcessor.from_pretrained('laion/larger_clap_general')
|
27 |
autoencoder = Autoencoder(autoencoder_path, 'stable_vae', quantization_first=True)
|
28 |
autoencoder.eval()
|