Spaces:
Running
on
Zero
Running
on
Zero
Update pnp.py
Browse files
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 |
-
|
79 |
-
|
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 |
-
|
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)
|