Spaces:
Runtime error
Runtime error
CSH-1220
commited on
Commit
Β·
2007ec0
1
Parent(s):
24363dc
Update filepath
Browse files
pipeline/morph_pipeline_successed_ver1.py
CHANGED
@@ -1140,7 +1140,7 @@ class AudioLDM2MorphPipeline(DiffusionPipeline,TextualInversionLoaderMixin):
|
|
1140 |
else:
|
1141 |
attn_procs[name] = AttnProcessor2_0()
|
1142 |
|
1143 |
-
state_dict = torch.load(
|
1144 |
for name, processor in attn_procs.items():
|
1145 |
if hasattr(processor, 'to_v_ip') or hasattr(processor, 'to_k_ip'):
|
1146 |
weight_name_v = name + ".to_v_ip.weight"
|
|
|
1140 |
else:
|
1141 |
attn_procs[name] = AttnProcessor2_0()
|
1142 |
|
1143 |
+
state_dict = torch.load(adapter_weight, map_location="cuda")
|
1144 |
for name, processor in attn_procs.items():
|
1145 |
if hasattr(processor, 'to_v_ip') or hasattr(processor, 'to_k_ip'):
|
1146 |
weight_name_v = name + ".to_v_ip.weight"
|