Skip to content

GitLab

Unverified Commit f5d3bb1d authored by Suraj Patil's avatar Suraj Patil Committed by GitHub
Browse files

fix CLIP conversion script (#13474)

parent 4be082ce
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/models/clip/convert_clip_original_pytorch_to_hf.py
View file @ f5d3bb1d
...@@ -117,7 +117,7 @@ def convert_clip_checkpoint(checkpoint_path, pytorch_dump_folder_path, config_pa ...@@ -117,7 +117,7 @@ def convert_clip_checkpoint(checkpoint_path, pytorch_dump_folder_path, config_pa
hf_model = CLIPModel(config).eval() 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() pt_model = pt_model.eval()
copy_text_model_and_projection(hf_model, pt_model) copy_text_model_and_projection(hf_model, pt_model)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment