Solved error in dataset.py.

This commit is contained in:
henri123lemoine committed 2023-03-24 22:21:29 -04:00
1 parent 581604b5b7
commit ea5bf2e707
4 files changed
+16 -12

No files matched your search

+1 -1
View File
@@ -6,7 +6,7 @@ COMPLETIONS_MODEL = "text-davinci-003"
LEN_EMBEDDINGS = 1536
MAX_LEN_PROMPT = 4095 # This may be 8191, unsure.
project_path = Path(__file__).parent.parent.parent
project_path = Path(__file__).parent.parent#.parent
PATH_TO_DATA = project_path / "src" / "data" / "alignment_texts.jsonl" # Path to the dataset .jsonl file.
PATH_TO_EMBEDDINGS = project_path / "src" / "data" / "embeddings.npy" # Path to the saved embeddings (.npy) file.
PATH_TO_DATASET = project_path / "src" / "data" / "dataset.pkl" # Path to the saved dataset (.pkl) file, containing the dataset class object.