Update model/openllama.py

model/openllama.py

@@ -171,7 +171,7 @@ class OpenLLAMAPEFTModel(nn.Module):
 
         self.visual_encoder, self.visual_hidden_size = imagebind_model.imagebind_huge(args)
         self.visual_encoder.to(self.device)
-        imagebind_ckpt = torch.load(imagebind_ckpt_path, map_location=self.device)
+        imagebind_ckpt = torch.load(imagebind_ckpt_path, map_location=torch.device('cuda'))
         self.visual_encoder.load_state_dict(imagebind_ckpt, strict=True)