Update models/llamaCustom.py
Browse files- models/llamaCustom.py +1 -2
models/llamaCustom.py
CHANGED
|
@@ -87,8 +87,7 @@ class LlamaCustom:
|
|
| 87 |
def __init__(self, model_name: str) -> None:
|
| 88 |
self.vector_index = self.initialize_index(model_name=model_name)
|
| 89 |
|
| 90 |
-
|
| 91 |
-
def initialize_index(_self, model_name: str):
|
| 92 |
index_name = model_name.split("/")[-1]
|
| 93 |
|
| 94 |
file_path = f"./vectorStores/{index_name}"
|
|
|
|
| 87 |
def __init__(self, model_name: str) -> None:
|
| 88 |
self.vector_index = self.initialize_index(model_name=model_name)
|
| 89 |
|
| 90 |
+
def initialize_index(self, model_name: str):
|
|
|
|
| 91 |
index_name = model_name.split("/")[-1]
|
| 92 |
|
| 93 |
file_path = f"./vectorStores/{index_name}"
|