feat : selective import

This commit is contained in:
martin legrand 2025-03-29 12:10:16 +01:00
parent ca50b1f2d0
commit de76fe14ea
2 changed files with 2 additions and 2 deletions

View File

@ -2,9 +2,9 @@
from .generator import GeneratorLLM from .generator import GeneratorLLM
class LlamacppLLM(GeneratorLLM): class LlamacppLLM(GeneratorLLM):
from llama_cpp import Llama
def __init__(self): def __init__(self):
from llama_cpp import Llama
""" """
Handle generation using llama.cpp Handle generation using llama.cpp
""" """

View File

@ -2,9 +2,9 @@
from .generator import GeneratorLLM from .generator import GeneratorLLM
class OllamaLLM(GeneratorLLM): class OllamaLLM(GeneratorLLM):
import ollama
def __init__(self): def __init__(self):
import ollama
""" """
Handle generation using Ollama. Handle generation using Ollama.
""" """