hbui commited on
Commit
8fe2b7c
·
1 Parent(s): 875cbe4

Update models/llamaCustom.py

Browse files
Files changed (1) hide show
  1. models/llamaCustom.py +3 -1
models/llamaCustom.py CHANGED
@@ -66,8 +66,10 @@ def load_model(model_name: str):
66
 
67
  class OurLLM(CustomLLM):
68
  def __init__(self, model_name: str, model_pipeline):
69
- self.model_name = model_name
70
  self.pipeline = model_pipeline
 
 
71
 
72
  @property
73
  def metadata(self) -> LLMMetadata:
 
66
 
67
  class OurLLM(CustomLLM):
68
  def __init__(self, model_name: str, model_pipeline):
69
+ #self.model_name = model_name
70
  self.pipeline = model_pipeline
71
+
72
+ super().__init__(model_name=model_name)
73
 
74
  @property
75
  def metadata(self) -> LLMMetadata: