Update clipGPT.py

clipGPT.py

@@ -148,7 +148,7 @@ def generate_caption_clipgpt(img):
 
     prefix_length = 10
     model = ClipGPT2Model(prefix_length, img_feature_size = 512)
-    model.load_state_dict(torch.load('model_train_best_run_clipGPT.pt'))
+    model.load_state_dict(torch.load('model_train_best_run_clipGPT.pt', map_location=torch.device('cpu')))
     model = model.eval()
     device=torch.device('cpu')
     model = model.to(device)