update llamacustom

models/llamaCustom.py

@@ -66,25 +66,10 @@ def load_model(model_name: str):
 
 
 class OurLLM(CustomLLM):
-    def __init__(self):
-        self.model_name = ""
-        self.pipeline = None
-
-    @property
-    def modelName(self):
-        return self.model_name
-
-    @property
-    def pipeline(self):
-        return self.pipeline
-
-    @modelName.setter
-    def modelName(self, name: str):
-        self.model_name = name
-
-    @pipeline.setter
-    def pipeline(self, value):
-        self.pipeline = value
+    def __init__(self, model_name: str, pipeline):
+        super().__init__()  # Call the __init__ method of CustomLLM
+        self.model_name = model_name
+        self.pipeline = pipeline
 
     @property
     def metadata(self) -> LLMMetadata:
@@ -138,9 +123,7 @@ class LlamaCustom:
 
             # define llm
             pipe = load_model(model_name=model_name)
-            llm = OurLLM()
-            llm.model_name = model_name
-            llm.pipeline = pipe
+            llm = OurLLM(model_name=model_name, pipeline=pipe)
 
             llm_predictor = LLMPredictor(llm=llm)
             service_context = ServiceContext.from_defaults(