Spaces:
Runtime error
Runtime error
Fixed bug aligning the names of the vector stores on train versus inference
Browse files
src/training/train_rag.py
CHANGED
@@ -33,8 +33,8 @@ def need_to_copy_vector_store(args) -> bool:
|
|
33 |
|
34 |
|
35 |
def copy_vector_store(args) -> None:
|
36 |
-
src = os.path.join(data_dir, 'vector_stores', args.in_vectors)
|
37 |
-
dest = os.path.join(data_dir, 'vector_stores', args.out_vectors)
|
38 |
shutil.copytree(src, dest)
|
39 |
|
40 |
|
|
|
33 |
|
34 |
|
35 |
def copy_vector_store(args) -> None:
|
36 |
+
src = os.path.join(data_dir, 'vector_stores', f"{args.in_vectors}_chroma")
|
37 |
+
dest = os.path.join(data_dir, 'vector_stores', f"{args.out_vectors}_chroma")
|
38 |
shutil.copytree(src, dest)
|
39 |
|
40 |
|