github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[pre-commit.ci] auto fixes from pre-commit.com hooks 

for more information, see https://pre-commit.ci
pull/6119/head
pre-commit-ci[bot] 2024-11-14 08:49:36 +00:00
parent 9995119c28
commit 38a7f3846d
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
applications/ColossalChat/coati/models/__init__.py
View File

@ -2,7 +2,7 @@ from .base import BaseModel
from .critic import Critic from .critic import Critic
from .generation import generate, generate_streaming, prepare_inputs_fn, update_model_kwargs_fn from .generation import generate, generate_streaming, prepare_inputs_fn, update_model_kwargs_fn
from .lora import LoraConfig, convert_to_lora_module, lora_manager from .lora import LoraConfig, convert_to_lora_module, lora_manager
from .loss import DpoLoss, KTOLoss, LogExpLoss, LogSigLoss, PolicyLoss, ValueLoss, PRMLoss from .loss import DpoLoss, KTOLoss, LogExpLoss, LogSigLoss, PolicyLoss, PRMLoss, ValueLoss
from .reward_model import RewardModel from .reward_model import RewardModel
from .utils import disable_dropout from .utils import disable_dropout
@ -24,5 +24,5 @@ __all__ = [
"disable_dropout", "disable_dropout",
"update_model_kwargs_fn", "update_model_kwargs_fn",
"prepare_inputs_fn", "prepare_inputs_fn",
"PRMLoss" "PRMLoss",
] ]