Skip to content

Commit 17b86cd

Browse files
authored
update modernvbert modeling (#404)
* update modernvbert modeling * lint * handle legacy * lint * changelog and pyproject
1 parent 9be8f19 commit 17b86cd

12 files changed

Lines changed: 16 additions & 764 deletions

CHANGELOG.md

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,14 @@ and this project adheres to [Semantic Versioning](http://semver.org/).
1111

1212
- Add ColQwen3.5 and BiQwen3.5 support (model + processor). Pretrained checkpoint: [athrael-soju/colqwen3.5-4.5B-v3](https://huggingface.co/athrael-soju/colqwen3.5-4.5B-v3).
1313

14+
### Changed
15+
16+
- Bump minimum supported `transformers` version to `5.3.0`.
17+
18+
### Fixed
19+
20+
- Fix ModernVBERT wrappers to rely on the upstream Hugging Face implementation and keep checkpoint key conversion mapping working with current Transformers v5 loading.
21+
1422
## [0.3.14] - 2026-02-24
1523

1624
### Added
Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,2 @@
1-
from .bivbert import BiModernVBert, BiModernVBertProcessor
2-
from .colvbert import ColModernVBert, ColModernVBertProcessor
1+
from .bimodernvbert import BiModernVBert, BiModernVBertProcessor
2+
from .colmodernvbert import ColModernVBert, ColModernVBertProcessor

colpali_engine/models/modernvbert/bivbert/__init__.py renamed to colpali_engine/models/modernvbert/bimodernvbert/__init__.py

File renamed without changes.

colpali_engine/models/modernvbert/bivbert/modeling_bimodernvbert.py renamed to colpali_engine/models/modernvbert/bimodernvbert/modeling_bimodernvbert.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,7 @@
11
from typing import Literal
22

33
import torch
4-
5-
from colpali_engine.models.modernvbert.modeling_modernvbert import ModernVBertModel, ModernVBertPreTrainedModel
4+
from transformers import ModernVBertModel, ModernVBertPreTrainedModel
65

76

87
class BiModernVBert(ModernVBertPreTrainedModel):

colpali_engine/models/modernvbert/bivbert/processing_bimodernvbert.py renamed to colpali_engine/models/modernvbert/bimodernvbert/processing_bimodernvbert.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
import torch
44
from transformers import BatchEncoding, BatchFeature
55

6-
from colpali_engine.models.modernvbert.colvbert import ColModernVBertProcessor # noqa: N801
6+
from colpali_engine.models.modernvbert.colmodernvbert import ColModernVBertProcessor # noqa: N801
77

88

99
class BiModernVBertProcessor(ColModernVBertProcessor): # noqa: N801

colpali_engine/models/modernvbert/colvbert/__init__.py renamed to colpali_engine/models/modernvbert/colmodernvbert/__init__.py

File renamed without changes.

colpali_engine/models/modernvbert/colvbert/modeling_colmodernvbert.py renamed to colpali_engine/models/modernvbert/colmodernvbert/modeling_colmodernvbert.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,12 @@
11
from torch import nn
2+
from transformers import ModernVBertModel, ModernVBertPreTrainedModel
23
from transformers.conversion_mapping import get_checkpoint_conversion_mapping, register_checkpoint_conversion_mapping
34
from transformers.core_model_loading import WeightRenaming
45

5-
from colpali_engine.models.modernvbert.modeling_modernvbert import ModernVBertModel, ModernVBertPreTrainedModel
6-
76

87
class ColModernVBert(ModernVBertPreTrainedModel):
98
_checkpoint_conversion_mapping = {
9+
r"^base_model\.model\.model\.text_model": "model.text_model",
1010
r"^base_model\.model\.custom_text_proj": "custom_text_proj",
1111
}
1212
"""

colpali_engine/models/modernvbert/colvbert/processing_colmodernvbert.py renamed to colpali_engine/models/modernvbert/colmodernvbert/processing_colmodernvbert.py

File renamed without changes.

colpali_engine/models/modernvbert/configuration_modernvbert.py

Lines changed: 0 additions & 279 deletions
This file was deleted.

0 commit comments

Comments
 (0)