Update LLM to extend from Pipeline

neuml · Dec 9, 2023 · 3dab4d8 · 3dab4d8
1 parent 6ce37d4
commit 3dab4d8
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/python/txtai/pipeline/llm/llm.py b/src/python/txtai/pipeline/llm/llm.py
@@ -4,8 +4,10 @@
 
 from .factory import GenerationFactory
 
+from ..base import Pipeline
 
-class LLM:
+
+class LLM(Pipeline):
     """
     Pipeline for running large language models (LLMs). This class supports the following LLM backends: