mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-06-18 19:58:17 +00:00
6ef7011462
* [chat] refactor lora * [chat] remove lm class * [chat] refactor save model * [chat] refactor train sft * [chat] fix ci * [chat] fix ci
9 lines
349 B
Python
9 lines
349 B
Python
from .base import Actor, Critic, RewardModel
|
|
from .lora import LoRAModule, convert_to_lora_module
|
|
from .loss import LogExpLoss, LogSigLoss, PolicyLoss, PPOPtxActorLoss, ValueLoss
|
|
|
|
__all__ = [
|
|
'Actor', 'Critic', 'RewardModel', 'PolicyLoss', 'ValueLoss', 'PPOPtxActorLoss', 'LogSigLoss', 'LogExpLoss',
|
|
'LoRAModule', 'convert_to_lora_module'
|
|
]
|