Spaces:
Sleeping
Sleeping
Zwea Htet commited on
Commit ·
3059501
1
Parent(s): b7600b8
update llamaCustom
Browse files- models/llamaCustom.py +1 -1
models/llamaCustom.py
CHANGED
|
@@ -66,7 +66,7 @@ def load_model(model_name: str):
|
|
| 66 |
return pipe
|
| 67 |
|
| 68 |
|
| 69 |
-
class OurLLM(CustomLLM
|
| 70 |
# def __init__(self, model_name: str, pipeline):
|
| 71 |
# super().__init__() # Call the __init__ method of CustomLLM
|
| 72 |
# self.model_name = model_name
|
|
|
|
| 66 |
return pipe
|
| 67 |
|
| 68 |
|
| 69 |
+
class OurLLM(CustomLLM):
|
| 70 |
# def __init__(self, model_name: str, pipeline):
|
| 71 |
# super().__init__() # Call the __init__ method of CustomLLM
|
| 72 |
# self.model_name = model_name
|