damerajee commited on
Commit
2da7b2f
·
verified ·
1 Parent(s): 5ef9619

Update modeling_gpt2vision.py

Browse files
Files changed (1) hide show
  1. modeling_gpt2vision.py +2 -2
modeling_gpt2vision.py CHANGED
@@ -4,11 +4,11 @@ from transformers import PreTrainedModel
4
  import re
5
 
6
  from .vision_encoder import VisionEncoder
7
- from .configuration_gpt2vision import GPTVisionConfig
8
  from .modeling_gpt2 import GPT2LMHeadModel
9
 
10
  class GPTVision(PreTrainedModel):
11
- config_class = GPTVisionConfig
12
 
13
  def __init__(self, config):
14
  super().__init__(config)
 
4
  import re
5
 
6
  from .vision_encoder import VisionEncoder
7
+ from .configuration_gpt2vision import GPT2VisionConfig
8
  from .modeling_gpt2 import GPT2LMHeadModel
9
 
10
  class GPTVision(PreTrainedModel):
11
+ config_class = GPT2VisionConfig
12
 
13
  def __init__(self, config):
14
  super().__init__(config)