Update llm/llamacpp/lc_model.py
Browse files- llm/llamacpp/lc_model.py +3 -3
llm/llamacpp/lc_model.py
CHANGED
@@ -8,7 +8,7 @@ from langchain.prompts import PromptTemplate
|
|
8 |
from langchain_community.llms import LlamaCpp
|
9 |
|
10 |
from llm.config import config
|
11 |
-
from llm.
|
12 |
|
13 |
print(os.getcwd())
|
14 |
|
@@ -31,7 +31,7 @@ print(path_to_models)
|
|
31 |
path_to_prompts = os.path.join(os.getcwd(), "/llm/")
|
32 |
print(path_to_prompts)
|
33 |
|
34 |
-
class LC_TinyLlama(
|
35 |
def __init__(self, prompt_entity: str, prompt_id: int = 0):
|
36 |
self.prompt_entity = prompt_entity
|
37 |
self.prompt_id = prompt_id
|
@@ -122,7 +122,7 @@ class LC_TinyLlama(LLMInterface, ABC):
|
|
122 |
return f"{self.__class__.__name__}(prompt_entity: {type(self.prompt_entity)} = {self.prompt_entity}, prompt_id: {type(self.prompt_id)} = {self.prompt_id})"
|
123 |
|
124 |
|
125 |
-
class LC_Phi3(
|
126 |
def __init__(self, prompt_entity: str, prompt_id: int = 0):
|
127 |
self.prompt_entity = prompt_entity
|
128 |
self.prompt_id = prompt_id
|
|
|
8 |
from langchain_community.llms import LlamaCpp
|
9 |
|
10 |
from llm.config import config
|
11 |
+
from llm.lc_interface import LCInterface
|
12 |
|
13 |
print(os.getcwd())
|
14 |
|
|
|
31 |
path_to_prompts = os.path.join(os.getcwd(), "/llm/")
|
32 |
print(path_to_prompts)
|
33 |
|
34 |
+
class LC_TinyLlama(LCInterface, ABC):
|
35 |
def __init__(self, prompt_entity: str, prompt_id: int = 0):
|
36 |
self.prompt_entity = prompt_entity
|
37 |
self.prompt_id = prompt_id
|
|
|
122 |
return f"{self.__class__.__name__}(prompt_entity: {type(self.prompt_entity)} = {self.prompt_entity}, prompt_id: {type(self.prompt_id)} = {self.prompt_id})"
|
123 |
|
124 |
|
125 |
+
class LC_Phi3(LCInterface, ABC):
|
126 |
def __init__(self, prompt_entity: str, prompt_id: int = 0):
|
127 |
self.prompt_entity = prompt_entity
|
128 |
self.prompt_id = prompt_id
|