riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

Fix image_processor_class bug (#21410)

Browse Source 
Co-authored-by: Shreshth Tuli <shreshthtuli@gmail.com>
This commit is contained in:
Shikhar Tuli 2023-02-02 09:20:52 -05:00 committed by GitHub
parent db572b3854
commit 0ae8dc0adf
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/commands/add_new_model_like.py
View File

@ -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? ",