S1131 commited on
Commit
188bee1
·
verified ·
1 Parent(s): 292fd4c

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -38,7 +38,7 @@ sys.path.append('/mount/src/gen_ai_dev')
38
  DATA_PATH = "./Infy financial report/"
39
  DATA_FILES = ["INFY_2022_2023.pdf", "INFY_2023_2024.pdf"]
40
  EMBEDDING_MODEL = "sentence-transformers/all-MiniLM-L6-v2"
41
- LLM_MODEL = "HuggingFaceH4/zephyr-7b-beta" # "microsoft/phi-2"
42
 
43
  # Environment settings
44
  os.environ["TOKENIZERS_PARALLELISM"] = "false"
 
38
  DATA_PATH = "./Infy financial report/"
39
  DATA_FILES = ["INFY_2022_2023.pdf", "INFY_2023_2024.pdf"]
40
  EMBEDDING_MODEL = "sentence-transformers/all-MiniLM-L6-v2"
41
+ LLM_MODEL = "gpt2" # "HuggingFaceH4/zephyr-7b-beta" # "microsoft/phi-2"
42
 
43
  # Environment settings
44
  os.environ["TOKENIZERS_PARALLELISM"] = "false"