map location=cpu

modules.py

@@ -381,7 +381,7 @@ class CLAP(nn.Module):
 
         self.logit_scale = nn.Parameter(torch.ones([]) * np.log(1 / 0.07))
 
-        state_dict = torch.load(ckpt_path)["model"]
+        state_dict = torch.load(ckpt_path, map_location="cpu")["model"]
         self.load_state_dict(self.clean_state_dict(state_dict))
         print("Loaded pretrained CLAP checkpoint.")