ColossalAI/applications/Chat/coati/models/llama/__init__.py

6 lines
156 B
Python

from .llama_actor import LlamaActor
from .llama_critic import LlamaCritic
from .llama_rm import LlamaRM
__all__ = ['LlamaActor', 'LlamaCritic', 'LlamaRM']