Spaces:
Runtime error
Runtime error
Tidied up space in formatting
Browse files
src/training/train_rag.py
CHANGED
@@ -14,7 +14,7 @@ from src.datatypes import *
|
|
14 |
|
15 |
|
16 |
# Set up and parse expected arguments to the script
|
17 |
-
parser=argparse.ArgumentParser(prog="train_rag",
|
18 |
description="Train a Chroma DB document store from the product dataset provided for use in a RAG LLM architecture")
|
19 |
parser.add_argument('-in_vectors', help="Optional existing Chroma DB to load and extend")
|
20 |
parser.add_argument('-out_vectors', required=True, help="The name for the output Chroma DB")
|
|
|
14 |
|
15 |
|
16 |
# Set up and parse expected arguments to the script
|
17 |
+
parser = argparse.ArgumentParser(prog="train_rag",
|
18 |
description="Train a Chroma DB document store from the product dataset provided for use in a RAG LLM architecture")
|
19 |
parser.add_argument('-in_vectors', help="Optional existing Chroma DB to load and extend")
|
20 |
parser.add_argument('-out_vectors', required=True, help="The name for the output Chroma DB")
|