Spaces:
Running
Running
File size: 223 Bytes
1b97239 |
1 2 3 4 5 |
from .model import ModelRegistry, LLaMAModel, OpenAIModel, AzureOpenAIModel
ModelRegistry.register("llama", LLaMAModel)
ModelRegistry.register("openai", OpenAIModel)
ModelRegistry.register("azure_openai", AzureOpenAIModel) |