update 14

run_image_captioning_flax_reduced.py

@@ -469,20 +469,15 @@ def main():
 
     # Load pretrained model and tokenizer
 
-    encoder_cache_dir, decoder_cache_dir = None, None
-    if model_args.cache_dir:
-        encoder_cache_dir = os.path.join(model_args.cache_dir, "encoder")
-        decoder_cache_dir = os.path.join(model_args.cache_dir, "decoder")
-
     # Use explicit specified encoder config
     if model_args.encoder_config_name:
         encoder_config = AutoConfig.from_pretrained(
-            model_args.encoder_config_name, cache_dir=encoder_cache_dir
+            model_args.encoder_config_name, cache_dir=model_args.cache_dir
         )
     # Use pretrained encoder model's config
     elif model_args.encoder_model_name_or_path:
         encoder_config = AutoConfig.from_pretrained(
-            model_args.encoder_model_name_or_path, cache_dir=encoder_cache_dir
+            model_args.encoder_model_name_or_path, cache_dir=model_args.cache_dir
         )
     else:
         raise ValueError(
@@ -492,12 +487,12 @@ def main():
     # Use explicit specified decoder config
     if model_args.decoder_config_name:
         decoder_config = AutoConfig.from_pretrained(
-            model_args.decoder_config_name, cache_dir=decoder_cache_dir
+            model_args.decoder_config_name, cache_dir=model_args.cache_dir
         )
     # Use pretrained decoder model's config
     elif model_args.decoder_model_name_or_path:
         decoder_config = AutoConfig.from_pretrained(
-            model_args.decoder_model_name_or_path, cache_dir=decoder_cache_dir
+            model_args.decoder_model_name_or_path, cache_dir=model_args.cache_dir
         )
     else:
         raise ValueError(