fix CLIP conversion script (#13474)

f5d3bb1d · Suraj Patil · GitHub · 4be082ce · f5d3bb1d
Unverified Commit f5d3bb1d authored Sep 08, 2021 by Suraj Patil Committed by GitHub Sep 08, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/models/clip/convert_clip_original_pytorch_to_hf.py ...ormers/models/clip/convert_clip_original_pytorch_to_hf.py +1 -1

No files found.
--- a/src/transformers/models/clip/convert_clip_original_pytorch_to_hf.py
+++ b/src/transformers/models/clip/convert_clip_original_pytorch_to_hf.py
@@ -117,7 +117,7 @@ def convert_clip_checkpoint(checkpoint_path, pytorch_dump_folder_path, config_pa

    hf_model = CLIPModel(config).eval()

-    pt_model, _ = load(checkpoint_path, jit=False)
+    pt_model, _ = load(checkpoint_path, device="cpu", jit=False)
    pt_model = pt_model.eval()

    copy_text_model_and_projection(hf_model, pt_model)