Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -25,6 +25,7 @@ from optimum.intel import (
|
|
| 25 |
OVModelForSequenceClassification,
|
| 26 |
OVModelForTokenClassification,
|
| 27 |
OVModelForPix2Struct,
|
|
|
|
| 28 |
OVWeightQuantizationConfig,
|
| 29 |
OVDiffusionPipeline,
|
| 30 |
)
|
|
@@ -39,6 +40,7 @@ _HEAD_TO_AUTOMODELS = {
|
|
| 39 |
"question-answering": "OVModelForQuestionAnswering",
|
| 40 |
"image-classification": "OVModelForImageClassification",
|
| 41 |
"audio-classification": "OVModelForAudioClassification",
|
|
|
|
| 42 |
}
|
| 43 |
|
| 44 |
|
|
|
|
| 25 |
OVModelForSequenceClassification,
|
| 26 |
OVModelForTokenClassification,
|
| 27 |
OVModelForPix2Struct,
|
| 28 |
+
OVModelForVisualCausalLM,
|
| 29 |
OVWeightQuantizationConfig,
|
| 30 |
OVDiffusionPipeline,
|
| 31 |
)
|
|
|
|
| 40 |
"question-answering": "OVModelForQuestionAnswering",
|
| 41 |
"image-classification": "OVModelForImageClassification",
|
| 42 |
"audio-classification": "OVModelForAudioClassification",
|
| 43 |
+
"image-text-to-text": "OVModelForVisualCausalLM"
|
| 44 |
}
|
| 45 |
|
| 46 |
|