Chengxu Zhuang commited on
Commit
bb62585
·
1 Parent(s): 2cd6995

minor fix for config class

Browse files
Files changed (1) hide show
  1. modeling_git.py +4 -1
modeling_git.py CHANGED
@@ -16,9 +16,12 @@ from transformers.models.opt.modeling_opt import OPTConfig
16
  import transformers.models.opt.modeling_opt as hg_opt
17
  import transformers.models.clip.modeling_clip as modeling_clip
18
  from transformers.modeling_outputs import SequenceClassifierOutputWithPast
 
19
 
20
 
21
  class GitForCausalLM(modeling_git.GitForCausalLM):
 
 
22
  def __init__(self, *args, **kwargs):
23
  super().__init__(*args, **kwargs)
24
 
@@ -219,4 +222,4 @@ class GitForSequenceClassification(modeling_git.GitPreTrainedModel):
219
  past_key_values=outputs.past_key_values,
220
  hidden_states=outputs.hidden_states,
221
  attentions=outputs.attentions,
222
- )
 
16
  import transformers.models.opt.modeling_opt as hg_opt
17
  import transformers.models.clip.modeling_clip as modeling_clip
18
  from transformers.modeling_outputs import SequenceClassifierOutputWithPast
19
+ from .configuration_git import GitConfig
20
 
21
 
22
  class GitForCausalLM(modeling_git.GitForCausalLM):
23
+ config_class = GitConfig
24
+
25
  def __init__(self, *args, **kwargs):
26
  super().__init__(*args, **kwargs)
27
 
 
222
  past_key_values=outputs.past_key_values,
223
  hidden_states=outputs.hidden_states,
224
  attentions=outputs.attentions,
225
+ )