Spaces:
Configuration error
Configuration error
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
|