Update edgeface-xxs/modeling_edgeface.py
Browse files
edgeface-xxs/modeling_edgeface.py
CHANGED
|
@@ -58,6 +58,7 @@ class EdgeFaceOutput(ModelOutput):
|
|
| 58 |
class EdgeFaceModel(PreTrainedModel):
|
| 59 |
config_class = EdgeFaceConfig
|
| 60 |
main_input_name = "pixel_values"
|
|
|
|
| 61 |
|
| 62 |
def __init__(self, config: EdgeFaceConfig):
|
| 63 |
super().__init__(config)
|
|
|
|
| 58 |
class EdgeFaceModel(PreTrainedModel):
|
| 59 |
config_class = EdgeFaceConfig
|
| 60 |
main_input_name = "pixel_values"
|
| 61 |
+
input_modalities="image"
|
| 62 |
|
| 63 |
def __init__(self, config: EdgeFaceConfig):
|
| 64 |
super().__init__(config)
|