Update model.py

model.py

@@ -115,8 +115,8 @@ class SALMONN(nn.Module):
             self.speech_Qformer.config.hidden_size, self.llama_model.config.hidden_size)
 
         # load ckpt
-        ckpt_dict = torch.load(ckpt)['model']
-        self.load_state_dict(ckpt_dict, strict=False, map_location=device)
+        ckpt_dict = torch.load(ckpt, map_location=device)['model']
+        self.load_state_dict(ckpt_dict, strict=False)
 
     def generate(
         self,