audioldm-text-to-audio-generation

Runtime error

haoheliu commited on Feb 2, 2023

Commit

ddc4da2

•

1 Parent(s): 412929c

update device

Files changed (1) hide show

audioldm/pipeline.py CHANGED Viewed

@@ -51,7 +51,7 @@ def build_model(config=None):
     resume_from_checkpoint = "./ckpt/ldm_trimmed.ckpt"
-    checkpoint = torch.load(resume_from_checkpoint)
     latent_diffusion.load_state_dict(checkpoint["state_dict"])
     latent_diffusion.eval()

     resume_from_checkpoint = "./ckpt/ldm_trimmed.ckpt"
+    checkpoint = torch.load(resume_from_checkpoint, map_location=device)
     latent_diffusion.load_state_dict(checkpoint["state_dict"])
     latent_diffusion.eval()