From bd8f45b1674be4da875097cb5557557d80bcce46 Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Wed, 8 Nov 2023 17:15:57 +0100 Subject: [PATCH] Fix tiny model script: not using `from_pt=True` (#27372) fix Co-authored-by: ydshieh --- utils/create_dummy_models.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/utils/create_dummy_models.py b/utils/create_dummy_models.py index 343a660200..5ee53ef7da 100644 --- a/utils/create_dummy_models.py +++ b/utils/create_dummy_models.py @@ -916,7 +916,7 @@ def build_composite_models(config_class, output_dir): model.save_pretrained(model_path) if tf_model_class is not None: - model = tf_model_class.from_pretrained(model_path, from_pt=True) + model = tf_model_class.from_pretrained(model_path) model.save_pretrained(model_path) # copy the processors @@ -1204,7 +1204,7 @@ def build(config_class, models_to_create, output_dir): ckpt = get_checkpoint_dir(output_dir, pt_arch) # Use the same weights from PyTorch. try: - model = tensorflow_arch.from_pretrained(ckpt, from_pt=True) + model = tensorflow_arch.from_pretrained(ckpt) model.save_pretrained(ckpt) except Exception as e: # Conversion may fail. Let's not create a model with different weights to avoid confusion (for now).