[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
This commit is contained in:
parent
c0cf5f257c
commit
b506fbd86f
1 changed files with 1 additions and 0 deletions
|
|
@ -1412,6 +1412,7 @@ class FastModel(FastBaseModel):
|
|||
if auto_model is None:
|
||||
if _num_labels is not None:
|
||||
from transformers import AutoModelForSequenceClassification
|
||||
|
||||
auto_model = AutoModelForSequenceClassification
|
||||
elif is_vlm:
|
||||
# Check if the model's auto_map supports the VLM auto class.
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue