SingingSDS / modules /llm /__init__.py
jhansss's picture
Fix Gemini LLM implementation and imports
6843cf8
raw
history blame
324 Bytes
from .base import AbstractLLMModel
from .registry import LLM_MODEL_REGISTRY, get_llm_model, register_llm_model
from .hf_pipeline import HFTextGenerationLLM
from .qwen import QwenLLM
from .gemini import GeminiLLM
__all__ = [
"AbstractLLMModel",
"get_llm_model",
"register_llm_model",
"LLM_MODEL_REGISTRY",
]