mirror of https://github.com/hpcaitech/ColossalAI
25 lines
658 B
Python
Executable File
25 lines
658 B
Python
Executable File
from .base import BaseModel
|
|
from .critic import Critic
|
|
from .generation import generate, generate_streaming, prepare_inputs_fn, update_model_kwargs_fn
|
|
from .lora import convert_to_lora_module
|
|
from .loss import DpoLoss, KTOLoss, LogExpLoss, LogSigLoss, PolicyLoss, ValueLoss
|
|
from .reward_model import RewardModel
|
|
from .utils import disable_dropout
|
|
|
|
__all__ = [
|
|
"BaseModel",
|
|
"Critic",
|
|
"RewardModel",
|
|
"PolicyLoss",
|
|
"ValueLoss",
|
|
"LogSigLoss",
|
|
"LogExpLoss",
|
|
"convert_to_lora_module",
|
|
"DpoLoss",
|
|
"KTOLoss" "generate",
|
|
"generate_streaming",
|
|
"disable_dropout",
|
|
"update_model_kwargs_fn",
|
|
"prepare_inputs_fn",
|
|
]
|