Skip to content

Commit

Permalink
add version to angle.config
Browse files Browse the repository at this point in the history
  • Loading branch information
SeanLee97 committed Aug 21, 2024
1 parent ddd52b8 commit 6fce33e
Show file tree
Hide file tree
Showing 4 changed files with 6 additions and 5 deletions.
2 changes: 1 addition & 1 deletion .bumpversion.cfg
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
[bumpversion]
files = setup.py angle_emb/__init__.py
files = setup.py angle_emb/version.py
current_version = 0.4.12
commit = True
tag = True
4 changes: 1 addition & 3 deletions angle_emb/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,4 @@

from .angle import * # NOQA
from .evaluation import * # NOQA


__version__ = '0.4.12'
from .version import __version__ # NOQA
4 changes: 3 additions & 1 deletion angle_emb/angle.py
Original file line number Diff line number Diff line change
Expand Up @@ -31,6 +31,7 @@
from .base import AngleBase
from .utils import logger
from .evaluation import CorrelationEvaluator
from .version import __version__


DEFAULT_LLM_PATTERNS = [r'.*llama.*', r'.*qwen.*', r'.*baichuan.*', r'.*mistral.*']
Expand Down Expand Up @@ -1114,7 +1115,7 @@ class AnglE(AngleBase):
:param load_mlm_model: bool. Whether load mlm model. Default False. If set True, it will load model with AutoModelForMaskedLM.
:param **kwargs: Any.
""" # NOQA
cfg_file_name = 'angle.config'
cfg_file_name = 'angle_config.json'
special_columns = ['labels', 'seperate_ids', 'extra', 'mask_target_labels']

def __init__(self,
Expand Down Expand Up @@ -1322,6 +1323,7 @@ def __init__(self,
'billm_model_class': billm_model_class,
'apply_lora': self.apply_lora,
'tokenizer_padding_side': tokenizer_padding_side,
'angle_emb_version': __version__,
}
self.__cfg.update(kwargs)

Expand Down
1 change: 1 addition & 0 deletions angle_emb/version.py
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
__version__ = '0.4.12'

0 comments on commit 6fce33e

Please sign in to comment.