github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/applications/Chat/coati/models/llama/__init__.py

6 lines
156 B
Python
Raw Normal View History

[Coati] first commit (#3283)
2023-03-28 12:25:36 +00:00
from .llama_actor import LlamaActor
from .llama_critic import LlamaCritic
from .llama_rm import LlamaRM
[chat] remove lm model class (#3653) * [chat] refactor lora * [chat] remove lm class * [chat] refactor save model * [chat] refactor train sft * [chat] fix ci * [chat] fix ci
2023-04-27 07:37:38 +00:00
__all__ = ['LlamaActor', 'LlamaCritic', 'LlamaRM']