Update app.py
Browse files
app.py
CHANGED
@@ -32,7 +32,7 @@ v_prediction = diff_config["ddim"]["v_prediction"]
|
|
32 |
|
33 |
processor = AutoProcessor.from_pretrained('laion/larger_clap_general')
|
34 |
clap_config = ClapConfig.from_pretrained("laion/larger_clap_general")
|
35 |
-
clapmodel = ClapModel(
|
36 |
clap_ckpt = torch.load(clap_bin_path, map_location='cpu')
|
37 |
clapmodel.load_state_dict(clap_ckpt)
|
38 |
clapmodel.to(device)
|
|
|
32 |
|
33 |
processor = AutoProcessor.from_pretrained('laion/larger_clap_general')
|
34 |
clap_config = ClapConfig.from_pretrained("laion/larger_clap_general")
|
35 |
+
clapmodel = ClapModel(clap_config)
|
36 |
clap_ckpt = torch.load(clap_bin_path, map_location='cpu')
|
37 |
clapmodel.load_state_dict(clap_ckpt)
|
38 |
clapmodel.to(device)
|