ccm commited on
Commit
06243da
·
verified ·
1 Parent(s): aa7a513

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -26,7 +26,7 @@ EXAMPLE_QUERIES = [
26
  "What is known about different modes for human-AI teaming?",
27
  ]
28
  EMBEDDING_MODEL_NAME = "allenai-specter"
29
- LLM_MODEL_NAME = "Qwen/Qwen2-7B-Instruct"
30
 
31
  # Load the dataset and convert to pandas
32
  data = pandas.read_parquet("hf://datasets/ccm/publications/data/train-00000-of-00001.parquet")
 
26
  "What is known about different modes for human-AI teaming?",
27
  ]
28
  EMBEDDING_MODEL_NAME = "allenai-specter"
29
+ LLM_MODEL_NAME = "Qwen/Qwen2-7B-Instruct-AWQ"
30
 
31
  # Load the dataset and convert to pandas
32
  data = pandas.read_parquet("hf://datasets/ccm/publications/data/train-00000-of-00001.parquet")