release v0.6.0

Former-commit-id: 51910d5803eb718e4976da0b3bfcdc5eeeea48eb
This commit is contained in:
hiyouga
2024-03-25 22:38:56 +08:00
parent ac2a401b1d
commit daab85e3e6
5 changed files with 8 additions and 9 deletions

View File

@@ -7,5 +7,5 @@ from .train import export_model, run_exp
from .webui import create_ui, create_web_demo
__version__ = "0.5.3"
__version__ = "0.6.0"
__all__ = ["create_app", "ChatModel", "Evaluator", "export_model", "run_exp", "create_ui", "create_web_demo"]

View File

@@ -45,7 +45,7 @@ class PairwiseTrainer(Trainer):
Subclass and override to inject custom behavior.
Note that the first element will be removed from the output tuple.
See: https://github.com/huggingface/transformers/blob/v4.30.2/src/transformers/trainer.py#L3509
See: https://github.com/huggingface/transformers/blob/v4.39.1/src/transformers/trainer.py#L3777
"""
# Compute rewards
_, _, values = model(**inputs, output_hidden_states=True, return_dict=True)