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',
|
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)
|
|
|
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)
|