Spaces:
Runtime error
Runtime error
Update models/llamaCustom.py
Browse files- 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:
|