From 3e0227d364fb8d1c1356aefda0d28857bccc6c2d 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 343a6602003487..5ee53ef7dab623 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).