hidehisa-arai commited on
Commit
119a1ed
1 Parent(s): 711b47e
Files changed (1) hide show
  1. configuration_japanese_clip.py +2 -2
configuration_japanese_clip.py CHANGED
@@ -1,5 +1,5 @@
1
 
2
- from transformers import PretrainedConfig, RobertaConfig
3
 
4
 
5
  class JapaneseCLIPVisionConfig(PretrainedConfig):
@@ -44,7 +44,7 @@ class JapaneseCLIPVisionConfig(PretrainedConfig):
44
  super().__init__(**kwargs)
45
 
46
 
47
- class JapaneseCLIPConfig(PretrainedConfig):
48
  model_type = "japanese_clip"
49
  is_composition = True
50
 
 
1
 
2
+ from transformers import PretrainedConfig, RobertaConfig, CLIPConfig
3
 
4
 
5
  class JapaneseCLIPVisionConfig(PretrainedConfig):
 
44
  super().__init__(**kwargs)
45
 
46
 
47
+ class JapaneseCLIPConfig(CLIPConfig):
48
  model_type = "japanese_clip"
49
  is_composition = True
50