/
githubmirror
/
ColossalAI
Обзор
Документация
Войти
/
githubmirror
/
ColossalAI
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
applications/ColossalChat/coati/models/__init__.py
28 строк
792 B
YeAnbang
Add GRPO and Support RLVR for PPO (#6186)
18 фев 2025, 04:43
Не верифицирован
18 фев 2025, 04:43
d20c8ff
Код
Авторство
О чём код?
from .base import BaseModel from .critic import Critic from .generation import generate, generate_streaming, prepare_inputs_fn, update_model_kwargs_fn from .lora import LoraConfig, convert_to_lora_module, lora_manager from .loss import DpoLoss, KTOLoss, LogExpLoss, LogSigLoss, PolicyLoss, ValueLoss from .reward_model import RewardModel from .rlvr_reward_model import RLVRRewardModel from .utils import disable_dropout __all__ = [ "BaseModel", "Critic", "RewardModel", "RLVRRewardModel", "PolicyLoss", "ValueLoss", "LogSigLoss", "LogExpLoss", "LoraConfig", "lora_manager", "convert_to_lora_module", "DpoLoss", "KTOLoss" "generate", "generate_streaming", "disable_dropout", "update_model_kwargs_fn", "prepare_inputs_fn", ]