Spaces:
Sleeping
Sleeping
Update models/llamaCustom.py
Browse files- models/llamaCustom.py +1 -1
models/llamaCustom.py
CHANGED
|
@@ -86,7 +86,7 @@ class LlamaCustom:
|
|
| 86 |
)
|
| 87 |
|
| 88 |
def __init__(self, model_name: str) -> None:
|
| 89 |
-
self.vector_index = self.initialize_index(
|
| 90 |
|
| 91 |
def initialize_index(self, model_name: str):
|
| 92 |
index_name = model_name.split("/")[-1]
|
|
|
|
| 86 |
)
|
| 87 |
|
| 88 |
def __init__(self, model_name: str) -> None:
|
| 89 |
+
self.vector_index = self.initialize_index(model_name=model_name)
|
| 90 |
|
| 91 |
def initialize_index(self, model_name: str):
|
| 92 |
index_name = model_name.split("/")[-1]
|