weights_only=True

s23_openai_clip.py

@@ -404,7 +404,7 @@ def get_image_embeddings(valid_df, model_path):
     valid_loader = build_loaders(valid_df, tokenizer, mode="valid")
 
     model = CLIPModel().to(CFG.device)
-    model.load_state_dict(torch.load(model_path, map_location=CFG.device))
+    model.load_state_dict(torch.load(model_path, map_location=CFG.device, weights_only=True))
     model.eval()
 
     valid_image_embeddings = []