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
......@@ -1556,6 +1556,8 @@ def get_user_input():
"What will be the name of the image processor class for this model? ",
default_value=f"{model_camel_cased}ImageProcessor",
)
else:
image_processor_class = None
if old_feature_extractor_class is not None:
feature_extractor_class = get_user_field(
"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