feat : selective import
This commit is contained in:
@@ -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
|
||||||
"""
|
"""
|
||||||
|
|||||||
@@ -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.
|
||||||
"""
|
"""
|
||||||
|
|||||||
Reference in New Issue
Block a user