Spaces:
Sleeping
Sleeping
burakcanbiner
commited on
Update pnp.py
Browse files
pnp.py
CHANGED
@@ -117,7 +117,7 @@ class PNP(nn.Module):
|
|
117 |
@spaces.GPU
|
118 |
def set_audio_projector(self, adapter_ckpt_path, audio_projector_ckpt_path):
|
119 |
|
120 |
-
print(f"SETTING MODEL TO {adapter_ckpt_path}")
|
121 |
gate_dict = torch.load(adapter_ckpt_path)
|
122 |
for name, param in self.unet.named_parameters():
|
123 |
if "adapter" in name:
|
@@ -240,7 +240,7 @@ class PNP(nn.Module):
|
|
240 |
|
241 |
# if not self.changed_model:
|
242 |
|
243 |
-
print(f"inside run_pnp {audio_projector_path}, {adapter_ckpt_path}")
|
244 |
|
245 |
self.set_audio_projector(adapter_ckpt_path, audio_projector_path)
|
246 |
|
|
|
117 |
@spaces.GPU
|
118 |
def set_audio_projector(self, adapter_ckpt_path, audio_projector_ckpt_path):
|
119 |
|
120 |
+
#print(f"SETTING MODEL TO {adapter_ckpt_path}")
|
121 |
gate_dict = torch.load(adapter_ckpt_path)
|
122 |
for name, param in self.unet.named_parameters():
|
123 |
if "adapter" in name:
|
|
|
240 |
|
241 |
# if not self.changed_model:
|
242 |
|
243 |
+
#print(f"inside run_pnp {audio_projector_path}, {adapter_ckpt_path}")
|
244 |
|
245 |
self.set_audio_projector(adapter_ckpt_path, audio_projector_path)
|
246 |
|