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

update init

pull/6119/head
Tong Li 2024-11-14 08:48:08 +00:00
parent 9ff9dc3d4a
commit 9995119c28
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
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 from .loss import DpoLoss, KTOLoss, LogExpLoss, LogSigLoss, PolicyLoss, ValueLoss, PRMLoss
from .reward_model import RewardModel from .reward_model import RewardModel
from .utils import disable_dropout from .utils import disable_dropout
@ -18,9 +18,11 @@ __all__ = [
"lora_manager", "lora_manager",
"convert_to_lora_module", "convert_to_lora_module",
"DpoLoss", "DpoLoss",
"KTOLoss" "generate", "KTOLoss",
"generate",
"generate_streaming", "generate_streaming",
"disable_dropout", "disable_dropout",
"update_model_kwargs_fn", "update_model_kwargs_fn",
"prepare_inputs_fn", "prepare_inputs_fn",
"PRMLoss"
] ]