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