Skip to content

GitLab

Unverified Commit 01562dac authored by amyeroberts's avatar amyeroberts Committed by GitHub
Browse files

Rename a class to reflect framework pattern AutoModelXxx -> TFAutoModelXxx (#16993)

parent 1be8d56e
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
src/transformers/models/auto/modeling_tf_auto.py
View file @ 01562dac
...@@ -411,7 +411,9 @@ class TFAutoModelForImageClassification(_BaseAutoModelClass): ...@@ -411,7 +411,9 @@ class TFAutoModelForImageClassification(_BaseAutoModelClass):
_model_mapping = TF_MODEL_FOR_IMAGE_CLASSIFICATION_MAPPING _model_mapping = TF_MODEL_FOR_IMAGE_CLASSIFICATION_MAPPING
AutoModelForImageClassification = auto_class_update(TFAutoModelForImageClassification, head_doc="image classification") TFAutoModelForImageClassification = auto_class_update(
TFAutoModelForImageClassification, head_doc="image classification"
)
class TFAutoModelForVision2Seq(_BaseAutoModelClass): class TFAutoModelForVision2Seq(_BaseAutoModelClass):
......
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