ydshieh commited on
Commit
b88266d
1 Parent(s): 199889f

fix model args

Browse files
vit_gpt2/modeling_flax_vit_gpt2_lm.py CHANGED
@@ -512,8 +512,8 @@ class FlaxViTGPT2LMForConditionalGeneration(FlaxViTGPT2LMPreTrainedModel):
512
  for key in text_kwargs.keys():
513
  del kwargs["text_" + key]
514
 
515
- vision_model_args = vision_kwargs.pop('model_args', None)
516
- text_model_args = text_kwargs.pop('model_args', None)
517
 
518
  # Load and initialize the vit & gpt2 model
519
  vision_model = vision_kwargs.pop("model", None)
 
512
  for key in text_kwargs.keys():
513
  del kwargs["text_" + key]
514
 
515
+ vision_model_args = vision_kwargs.pop('model_args', [])
516
+ text_model_args = text_kwargs.pop('model_args', [])
517
 
518
  # Load and initialize the vit & gpt2 model
519
  vision_model = vision_kwargs.pop("model", None)