modification

DiseaseSpecific/utils.py

@@ -77,7 +77,7 @@ def load_model(model_path, args, n_ent, n_rel, device):
     for key, size, count in params:
         logger.info('Key:{0}, Size:{1}, Count:{2}'.format(key, size, count))
         
-    model.load_state_dict(model_params)
+    model.load_state_dict(model_params, map_location=device)
     model.eval()
     logger.info(model)
     

server.py

@@ -12,7 +12,8 @@ import spacy
 # os.system("python -m spacy download en-core-web-sm")
 import pickle as pkl
 #%%
-
+# please use torch.load with map_location=torch.device('cpu') to map your storages to the CPU.
+# torch.loa
 from torch.nn.modules.loss import CrossEntropyLoss
 from transformers import AutoTokenizer
 from transformers import BioGptForCausalLM, BartForConditionalGeneration