camenduru
/

audioldm

camenduru commited on Feb 2, 2023

Commit

2075399

1 Parent(s): 120a250

Update audioldm/pipeline.py

Files changed (1) hide show

audioldm/pipeline.py CHANGED Viewed

@@ -49,7 +49,7 @@ def build_model(config=None):
     # No normalization here
     latent_diffusion = LatentDiffusion(**config["model"]["params"])
-    resume_from_checkpoint = "./ckpt/ldm_trimmed.ckpt"
     checkpoint = torch.load(resume_from_checkpoint, map_location=device)
     latent_diffusion.load_state_dict(checkpoint["state_dict"])

     # No normalization here
     latent_diffusion = LatentDiffusion(**config["model"]["params"])
+    resume_from_checkpoint = "/content/audioldm/ckpt/ldm_trimmed.ckpt"
     checkpoint = torch.load(resume_from_checkpoint, map_location=device)
     latent_diffusion.load_state_dict(checkpoint["state_dict"])