Update llm/llm_factory.py
Browse files- llm/llm_factory.py +2 -0
llm/llm_factory.py
CHANGED
@@ -10,6 +10,8 @@ class LLM_Factory:
|
|
10 |
@staticmethod
|
11 |
def create_llm(prompt_entity: str, prompt_id: int, trigger: Dict[str, str]):
|
12 |
|
|
|
|
|
13 |
if trigger.keys() == "hf" and trigger.values() == "effective":
|
14 |
print(trigger)
|
15 |
model = HF_Mistaril(prompt_entity=prompt_entity, prompt_id=prompt_id)
|
|
|
10 |
@staticmethod
|
11 |
def create_llm(prompt_entity: str, prompt_id: int, trigger: Dict[str, str]):
|
12 |
|
13 |
+
print(trigger)
|
14 |
+
|
15 |
if trigger.keys() == "hf" and trigger.values() == "effective":
|
16 |
print(trigger)
|
17 |
model = HF_Mistaril(prompt_entity=prompt_entity, prompt_id=prompt_id)
|