Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
wangsen
MinerU
Commits
60ff0c39
Commit
60ff0c39
authored
Jun 13, 2025
by
myhloli
Browse files
feat: integrate automatic model path retrieval in model.py
parent
3dd9a832
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
0 deletions
+4
-0
mineru/model/vlm_sglang_model/model.py
mineru/model/vlm_sglang_model/model.py
+4
-0
No files found.
mineru/model/vlm_sglang_model/model.py
View file @
60ff0c39
...
@@ -22,6 +22,7 @@ from transformers import (
...
@@ -22,6 +22,7 @@ from transformers import (
from
..vlm_hf_model.configuration_mineru2
import
Mineru2QwenConfig
from
..vlm_hf_model.configuration_mineru2
import
Mineru2QwenConfig
from
..vlm_hf_model.modeling_mineru2
import
build_vision_projector
from
..vlm_hf_model.modeling_mineru2
import
build_vision_projector
from
...utils.models_download_utils
import
auto_download_and_get_model_root_path
def
flatten_nested_list
(
nested_list
):
def
flatten_nested_list
(
nested_list
):
...
@@ -61,6 +62,9 @@ class Mineru2QwenForCausalLM(nn.Module):
...
@@ -61,6 +62,9 @@ class Mineru2QwenForCausalLM(nn.Module):
# load vision tower
# load vision tower
mm_vision_tower
=
self
.
config
.
mm_vision_tower
mm_vision_tower
=
self
.
config
.
mm_vision_tower
model_root_path
=
auto_download_and_get_model_root_path
(
"/"
,
"vlm"
)
mm_vision_tower
=
f
"
{
model_root_path
}
/
{
mm_vision_tower
}
"
if
"clip"
in
mm_vision_tower
:
if
"clip"
in
mm_vision_tower
:
vision_config
=
CLIPVisionConfig
.
from_pretrained
(
mm_vision_tower
)
vision_config
=
CLIPVisionConfig
.
from_pretrained
(
mm_vision_tower
)
self
.
vision_tower
=
CLIPVisionModel
(
vision_config
)
# type: ignore
self
.
vision_tower
=
CLIPVisionModel
(
vision_config
)
# type: ignore
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment