Spaces:

burakcanbiner
/

SonicDiffusion

Running on Zero

burakcanbiner commited on May 10, 2024

Commit

d35aa53

verified ·

1 Parent(s): fb2991d

Update pnp.py

Files changed (1) hide show

pnp.py CHANGED Viewed

@@ -108,7 +108,7 @@ class PNP(nn.Module):
         # self.set_audio_projector(adapter_ckpt_path, audio_projector_ckpt_path)
         self.text_encoder = self.text_encoder.cuda()
-        self.audio_projector.load_state_dict(torch.load(audio_projector_path))
         self.audio_projector_ckpt_path = audio_projector_ckpt_path
         self.adapter_ckpt_path = adapter_ckpt_path

         # self.set_audio_projector(adapter_ckpt_path, audio_projector_ckpt_path)
         self.text_encoder = self.text_encoder.cuda()
+        #self.audio_projector.load_state_dict(torch.load(audio_projector_path))
         self.audio_projector_ckpt_path = audio_projector_ckpt_path
         self.adapter_ckpt_path = adapter_ckpt_path