From de76fe14ea31d24b5c31a2f8777276b4521dfbe1 Mon Sep 17 00:00:00 2001 From: martin legrand Date: Sat, 29 Mar 2025 12:10:16 +0100 Subject: [PATCH] feat : selective import --- server/sources/llamacpp.py | 2 +- server/sources/ollama.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/server/sources/llamacpp.py b/server/sources/llamacpp.py index 524cb1d..09fd465 100644 --- a/server/sources/llamacpp.py +++ b/server/sources/llamacpp.py @@ -2,9 +2,9 @@ from .generator import GeneratorLLM class LlamacppLLM(GeneratorLLM): - from llama_cpp import Llama def __init__(self): + from llama_cpp import Llama """ Handle generation using llama.cpp """ diff --git a/server/sources/ollama.py b/server/sources/ollama.py index 548a1d8..8cd7631 100644 --- a/server/sources/ollama.py +++ b/server/sources/ollama.py @@ -2,9 +2,9 @@ from .generator import GeneratorLLM class OllamaLLM(GeneratorLLM): - import ollama def __init__(self): + import ollama """ Handle generation using Ollama. """