update test_model.py

tests/test_model.py

@@ -25,9 +25,12 @@ max_length = 8
 vision_model_name = 'google/vit-base-patch16-224-in21k'
 text_model_name = 'asi/gpt-fr-cased-small'
 
+project_encoder = False
+
 flax_vit_gpt2_lm = FlaxViTGPT2LMForConditionalGeneration.from_vision_text_pretrained(
     vision_pretrained_model_name_or_path=vision_model_name,
-    text_pretrained_model_name_or_path=text_model_name
+    text_pretrained_model_name_or_path=text_model_name,
+    project_encoder=project_encoder
 )
 model = flax_vit_gpt2_lm
 
@@ -103,6 +106,8 @@ print('=' * 60)
 print(f'orig. GPT2 generated caption: {orig_caption}')
 print(f'GPT2 generated caption: {caption}')
 
+assert list(orig_token_ids) == list(token_ids)
+
 # model data
 model_inputs = {
     'pixel_values': pixel_values,