Update requirements and make necessary code changes

.ipynb_checkpoints/model_loader-checkpoint.py

@@ -634,7 +634,7 @@ def load_model():
     local_file = hf_hub_download(repo_id=checkpoint, filename="cpt.pth")
 
     # Load the best model state
-    state_dict = torch.load(local_file, weights_only=True)
+    state_dict = torch.load(local_file, map_location=torch.device('cpu'), weights_only=True)
     model.load_state_dict(state_dict)
 
     return model, tokenizer

model_loader.py

@@ -634,7 +634,7 @@ def load_model():
     local_file = hf_hub_download(repo_id=checkpoint, filename="cpt.pth")
 
     # Load the best model state
-    state_dict = torch.load(local_file, weights_only=True)
+    state_dict = torch.load(local_file, map_location=torch.device('cpu'), weights_only=True)
     model.load_state_dict(state_dict)
 
     return model, tokenizer