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]
|