Update pnp.py

pnp.py

@@ -68,15 +68,15 @@ class PNP(nn.Module):
         
 
 
-        audio_projector_path = "ckpts/audio_projector_landscape.pth"
-        adapter_ckpt_path = "ckpts/landscape.pt"    
+        self.audio_projector_path = "ckpts/audio_projector_landscape.pth"
+        self.adapter_ckpt_path = "ckpts/landscape.pt"    
         #self.pnp.set_audio_projector(gate_dict_path, audio_projector_path)   
                     
-        gate_dict = torch.load(adapter_ckpt_path)
+        # gate_dict = torch.load(adapter_ckpt_path)
         
-        for name, param in self.unet.named_parameters():
-           if "adapter" in name:
-            param.data = gate_dict[name]
+        # for name, param in self.unet.named_parameters():
+        #    if "adapter" in name:
+        #     param.data = gate_dict[name]
                
         #unet.to(self.device);
         
@@ -237,8 +237,12 @@ class PNP(nn.Module):
                 cfg_scale=5):
             
 #         if not self.changed_model:
-#             self.set_audio_projector(self.adapter_ckpt_path, self.audio_projector_ckpt_path)    
-        
+
+        audio_projector_path = "ckpts/audio_projector_landscape.pth"
+        adapter_ckpt_path = "ckpts/landscape.pt" 
+        self.set_audio_projector(adapter_ckpt_path, audio_projector_path)    
+
+                    
         self.audio_projector = self.audio_projector.cuda()
         
         self.set_text_embeds(prompt)