SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

fix loading clip vision model (#12566)

Browse Source
This commit is contained in:
Suraj Patil
2021-07-07 22:50:27 +05:30
committed by GitHub
parent b86826099b
commit d7e156bd1a
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/research_projects/jax-projects/hybrid_clip/configuration_hybrid_clip.py
View File

@@ -75,6 +75,10 @@ class HybridCLIPConfig(PretrainedConfig):
if vision_model_type == "clip": if vision_model_type == "clip":
self.vision_config = AutoConfig.for_model(vision_model_type, **vision_config).vision_config self.vision_config = AutoConfig.for_model(vision_model_type, **vision_config).vision_config
elif vision_model_type == "clip_vision_model":
from transformers import CLIPVisionConfig
self.vision_config = CLIPVisionConfig(**vision_config)
else: else:
self.vision_config = AutoConfig.for_model(vision_model_type, **vision_config) self.vision_config = AutoConfig.for_model(vision_model_type, **vision_config)