passionMan commited on
Commit
aee0e28
·
verified ·
1 Parent(s): 6983777

Add BrainOCR model code: configuration_brain_ocr.py

Browse files
Files changed (1) hide show
  1. configuration_brain_ocr.py +2 -2
configuration_brain_ocr.py CHANGED
@@ -6,7 +6,7 @@ from transformers import PretrainedConfig
6
 
7
 
8
  class BrainOCRVisionConfig(PretrainedConfig):
9
- model_type = "brain_ocr"
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 = "brain_ocr"
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,