mirror of https://github.com/hpcaitech/ColossalAI
8 lines
157 B
Python
8 lines
157 B
Python
|
from .llama import LlamaModelInferPolicy
|
||
|
|
|
||
|
|
model_policy_map = {
|
||
|
|
"llama": LlamaModelInferPolicy,
|
||
|
|
}
|
||
|
|
|
||
|
|
__all__ = ["LlamaModelInferPolicy", "model_polic_map"]
|