hbui commited on
Commit
9d321e7
·
1 Parent(s): 8fe2b7c

Update models/llamaCustom.py

Browse files
Files changed (1) hide show
  1. models/llamaCustom.py +2 -1
models/llamaCustom.py CHANGED
@@ -67,9 +67,10 @@ def load_model(model_name: str):
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:
 
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
+ super().__init__(pipeline=model_pipeline)
74
 
75
  @property
76
  def metadata(self) -> LLMMetadata: