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('/Data/home/Dennis/DeepMIR-2024/Final_Project/AP-adapter/pytorch_model.bin', 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"
 
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"