update llamaCustom

models/llamaCustom.py

@@ -66,7 +66,7 @@ def load_model(model_name: str):
     return pipe
 
 
-class OurLLM(CustomLLM, BaseModel):
+class OurLLM(CustomLLM):
     # def __init__(self, model_name: str, pipeline):
     #     super().__init__()  # Call the __init__ method of CustomLLM
     #     self.model_name = model_name