S1131 commited on
Commit
50be3c7
·
verified ·
1 Parent(s): 711421f

microsoft/phi-1_5

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