From f5d3bb1dd20e9d2c3c5bd66b30032bea6c3f45d8 Mon Sep 17 00:00:00 2001 From: Suraj Patil Date: Wed, 8 Sep 2021 12:57:18 +0530 Subject: [PATCH] fix CLIP conversion script (#13474) --- .../models/clip/convert_clip_original_pytorch_to_hf.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/models/clip/convert_clip_original_pytorch_to_hf.py b/src/transformers/models/clip/convert_clip_original_pytorch_to_hf.py index fdd4c148a94..58886aa88a3 100644 --- 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)