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