burakcanbiner commited on
Commit
0667969
1 Parent(s): 07d248b

Update pnp.py

Browse files
Files changed (1) hide show
  1. pnp.py +2 -2
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