Spaces:
Sleeping
Sleeping
Update pnp.py
Browse files
pnp.py
CHANGED
@@ -234,12 +234,12 @@ class PNP(nn.Module):
|
|
234 |
def run_pnp(self, n_timesteps=50, pnp_f_t=0.5, pnp_attn_t=0.5,
|
235 |
prompt="", negative_prompt="",
|
236 |
audio_path="", image_path="",
|
|
|
|
|
237 |
cfg_scale=5):
|
238 |
|
239 |
# if not self.changed_model:
|
240 |
|
241 |
-
audio_projector_path = "ckpts/audio_projector_landscape.pth"
|
242 |
-
adapter_ckpt_path = "ckpts/landscape.pt"
|
243 |
self.set_audio_projector(adapter_ckpt_path, audio_projector_path)
|
244 |
|
245 |
|
|
|
234 |
def run_pnp(self, n_timesteps=50, pnp_f_t=0.5, pnp_attn_t=0.5,
|
235 |
prompt="", negative_prompt="",
|
236 |
audio_path="", image_path="",
|
237 |
+
audio_projector_path = "ckpts/audio_projector_landscape.pth",
|
238 |
+
adapter_ckpt_path = "ckpts/landscape.pt",
|
239 |
cfg_scale=5):
|
240 |
|
241 |
# if not self.changed_model:
|
242 |
|
|
|
|
|
243 |
self.set_audio_projector(adapter_ckpt_path, audio_projector_path)
|
244 |
|
245 |
|