Skip to content

GitLab

Unverified Commit 0ae8dc0a authored by Shikhar Tuli's avatar Shikhar Tuli Committed by GitHub
Browse files

Fix image_processor_class bug (#21410) 


Co-authored-by: default avatarShreshth Tuli <shreshthtuli@gmail.com>
parent db572b38
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
src/transformers/commands/add_new_model_like.py
View file @ 0ae8dc0a
...@@ -1556,6 +1556,8 @@ def get_user_input(): ...@@ -1556,6 +1556,8 @@ def get_user_input():
"What will be the name of the image processor class for this model? ", "What will be the name of the image processor class for this model? ",
default_value=f"{model_camel_cased}ImageProcessor", default_value=f"{model_camel_cased}ImageProcessor",
) )
else:
image_processor_class = None
if old_feature_extractor_class is not None: if old_feature_extractor_class is not None:
feature_extractor_class = get_user_field( feature_extractor_class = get_user_field(
"What will be the name of the feature extractor class for this model? ", "What will be the name of the feature extractor class for this model? ",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment