Spaces:
Sleeping
Sleeping
Update pnp.py
Browse files
pnp.py
CHANGED
@@ -108,7 +108,7 @@ class PNP(nn.Module):
|
|
108 |
# self.set_audio_projector(adapter_ckpt_path, audio_projector_ckpt_path)
|
109 |
self.text_encoder = self.text_encoder.cuda()
|
110 |
|
111 |
-
self.audio_projector.load_state_dict(torch.load(audio_projector_path))
|
112 |
|
113 |
self.audio_projector_ckpt_path = audio_projector_ckpt_path
|
114 |
self.adapter_ckpt_path = adapter_ckpt_path
|
|
|
108 |
# self.set_audio_projector(adapter_ckpt_path, audio_projector_ckpt_path)
|
109 |
self.text_encoder = self.text_encoder.cuda()
|
110 |
|
111 |
+
#self.audio_projector.load_state_dict(torch.load(audio_projector_path))
|
112 |
|
113 |
self.audio_projector_ckpt_path = audio_projector_ckpt_path
|
114 |
self.adapter_ckpt_path = adapter_ckpt_path
|