Update llm/huggingfacehub/hf_model.py
Browse files
llm/huggingfacehub/hf_model.py
CHANGED
|
@@ -48,6 +48,7 @@ class HF_Mistaril(LLMInterface, ABC):
|
|
| 48 |
data = yaml.safe_load(file)
|
| 49 |
return data
|
| 50 |
except Exception as e:
|
|
|
|
| 51 |
logger.error(msg="Execution filed", exc_info=e)
|
| 52 |
|
| 53 |
def execution(self):
|
|
@@ -61,6 +62,7 @@ class HF_Mistaril(LLMInterface, ABC):
|
|
| 61 |
output = llm_chain.invoke(self.prompt_entity)
|
| 62 |
return output["text"]
|
| 63 |
except Exception as e:
|
|
|
|
| 64 |
logger.critical(msg="Execution filed", exc_info=e)
|
| 65 |
|
| 66 |
def __str__(self):
|
|
@@ -91,6 +93,7 @@ class HF_TinyLlama(LLMInterface, ABC):
|
|
| 91 |
data = yaml.safe_load(file)
|
| 92 |
return data
|
| 93 |
except Exception as e:
|
|
|
|
| 94 |
logger.error(msg="Execution filed", exc_info=e)
|
| 95 |
|
| 96 |
def execution(self):
|
|
@@ -104,6 +107,7 @@ class HF_TinyLlama(LLMInterface, ABC):
|
|
| 104 |
output = llm_chain.invoke(self.prompt_entity)
|
| 105 |
return output["text"]
|
| 106 |
except Exception as e:
|
|
|
|
| 107 |
logger.critical(msg="Execution filed", exc_info=e)
|
| 108 |
|
| 109 |
def __str__(self):
|
|
|
|
| 48 |
data = yaml.safe_load(file)
|
| 49 |
return data
|
| 50 |
except Exception as e:
|
| 51 |
+
print(f"Execution filed : {e}")
|
| 52 |
logger.error(msg="Execution filed", exc_info=e)
|
| 53 |
|
| 54 |
def execution(self):
|
|
|
|
| 62 |
output = llm_chain.invoke(self.prompt_entity)
|
| 63 |
return output["text"]
|
| 64 |
except Exception as e:
|
| 65 |
+
print(f"Execution filed : {e}")
|
| 66 |
logger.critical(msg="Execution filed", exc_info=e)
|
| 67 |
|
| 68 |
def __str__(self):
|
|
|
|
| 93 |
data = yaml.safe_load(file)
|
| 94 |
return data
|
| 95 |
except Exception as e:
|
| 96 |
+
print(f"Execution filed : {e}")
|
| 97 |
logger.error(msg="Execution filed", exc_info=e)
|
| 98 |
|
| 99 |
def execution(self):
|
|
|
|
| 107 |
output = llm_chain.invoke(self.prompt_entity)
|
| 108 |
return output["text"]
|
| 109 |
except Exception as e:
|
| 110 |
+
print(f"Execution filed : {e}")
|
| 111 |
logger.critical(msg="Execution filed", exc_info=e)
|
| 112 |
|
| 113 |
def __str__(self):
|