burakcanbiner commited on
Commit
127b8a4
·
verified ·
1 Parent(s): d35aa53

Update pnp.py

Browse files
Files changed (1) hide show
  1. pnp.py +12 -8
pnp.py CHANGED
@@ -68,15 +68,15 @@ class PNP(nn.Module):
68
 
69
 
70
 
71
- audio_projector_path = "ckpts/audio_projector_landscape.pth"
72
- adapter_ckpt_path = "ckpts/landscape.pt"
73
  #self.pnp.set_audio_projector(gate_dict_path, audio_projector_path)
74
 
75
- gate_dict = torch.load(adapter_ckpt_path)
76
 
77
- for name, param in self.unet.named_parameters():
78
- if "adapter" in name:
79
- param.data = gate_dict[name]
80
 
81
  #unet.to(self.device);
82
 
@@ -237,8 +237,12 @@ class PNP(nn.Module):
237
  cfg_scale=5):
238
 
239
  # if not self.changed_model:
240
- # self.set_audio_projector(self.adapter_ckpt_path, self.audio_projector_ckpt_path)
241
-
 
 
 
 
242
  self.audio_projector = self.audio_projector.cuda()
243
 
244
  self.set_text_embeds(prompt)
 
68
 
69
 
70
 
71
+ self.audio_projector_path = "ckpts/audio_projector_landscape.pth"
72
+ self.adapter_ckpt_path = "ckpts/landscape.pt"
73
  #self.pnp.set_audio_projector(gate_dict_path, audio_projector_path)
74
 
75
+ # gate_dict = torch.load(adapter_ckpt_path)
76
 
77
+ # for name, param in self.unet.named_parameters():
78
+ # if "adapter" in name:
79
+ # param.data = gate_dict[name]
80
 
81
  #unet.to(self.device);
82
 
 
237
  cfg_scale=5):
238
 
239
  # if not self.changed_model:
240
+
241
+ audio_projector_path = "ckpts/audio_projector_landscape.pth"
242
+ adapter_ckpt_path = "ckpts/landscape.pt"
243
+ self.set_audio_projector(adapter_ckpt_path, audio_projector_path)
244
+
245
+
246
  self.audio_projector = self.audio_projector.cuda()
247
 
248
  self.set_text_embeds(prompt)