Add BrainOCR model code: configuration_brain_ocr.py
Browse files
configuration_brain_ocr.py
CHANGED
|
@@ -6,7 +6,7 @@ from transformers import PretrainedConfig
|
|
| 6 |
|
| 7 |
|
| 8 |
class BrainOCRVisionConfig(PretrainedConfig):
|
| 9 |
-
model_type = "
|
| 10 |
base_config_key = "vision_config"
|
| 11 |
|
| 12 |
def __init__(
|
|
@@ -196,7 +196,7 @@ class BrainOCRTextConfig(PretrainedConfig):
|
|
| 196 |
|
| 197 |
|
| 198 |
class BrainOCRConfig(PretrainedConfig):
|
| 199 |
-
model_type = "
|
| 200 |
sub_configs = {
|
| 201 |
"vision_config": BrainOCRVisionConfig,
|
| 202 |
"text_config": BrainOCRTextConfig,
|
|
|
|
| 6 |
|
| 7 |
|
| 8 |
class BrainOCRVisionConfig(PretrainedConfig):
|
| 9 |
+
model_type = "hunyuan_vl"
|
| 10 |
base_config_key = "vision_config"
|
| 11 |
|
| 12 |
def __init__(
|
|
|
|
| 196 |
|
| 197 |
|
| 198 |
class BrainOCRConfig(PretrainedConfig):
|
| 199 |
+
model_type = "hunyuan_vl"
|
| 200 |
sub_configs = {
|
| 201 |
"vision_config": BrainOCRVisionConfig,
|
| 202 |
"text_config": BrainOCRTextConfig,
|