CineAI commited on
Commit
fa98157
1 Parent(s): f12e651

Update llm/llamacpp/lc_model.py

Browse files
Files changed (1) hide show
  1. 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.llm_interface import LLMInterface
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(LLMInterface, 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,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(LLMInterface, ABC):
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