Tao Wu commited on
Commit
babde94
·
1 Parent(s): 3310e88
Files changed (2) hide show
  1. app/config.py +1 -1
  2. app/embedding_setup.py +1 -1
app/config.py CHANGED
@@ -16,7 +16,7 @@ ENCODE_KWARGS = {
16
  QUERY_INSTRUCTION = os.getenv('QUERY_INSTRUCTION', '')
17
 
18
  # Other configurations
19
- TOP_K = int(os.getenv('TOP_K', 5))
20
  #PERSIST_DIRECTORY = os.getenv('PERSIST_DIRECTORY', "/app/data/course_emb_db")
21
  PERSIST_DIRECTORY = os.getenv('PERSIST_DIRECTORY', "/app/data/BA_course")
22
  CSV_FILE_PATH = os.getenv('CSV_FILE_PATH', '/app/data/occupations_de.csv')
 
16
  QUERY_INSTRUCTION = os.getenv('QUERY_INSTRUCTION', '')
17
 
18
  # Other configurations
19
+ TOP_K = int(os.getenv('TOP_K', 10))
20
  #PERSIST_DIRECTORY = os.getenv('PERSIST_DIRECTORY', "/app/data/course_emb_db")
21
  PERSIST_DIRECTORY = os.getenv('PERSIST_DIRECTORY', "/app/data/BA_course")
22
  CSV_FILE_PATH = os.getenv('CSV_FILE_PATH', '/app/data/occupations_de.csv')
app/embedding_setup.py CHANGED
@@ -31,7 +31,7 @@ retriever = db.as_retriever(search_kwargs={"k": TOP_K})
31
 
32
 
33
  LLM_MODEL = "meta-llama/Meta-Llama-3-8B-Instruct"
34
- lora_weights = "/hpcwork/vg380347/llama3/Instruct_8B_EngGer_alpaca_finetune_pairwise_skill_24_128/last_checkpoint"
35
 
36
  hf_auth = os.environ.get("hf_token")
37
 
 
31
 
32
 
33
  LLM_MODEL = "meta-llama/Meta-Llama-3-8B-Instruct"
34
+ lora_weights = "wt3639/Llama-3-8B-Instruct_CourseRec_lora"
35
 
36
  hf_auth = os.environ.get("hf_token")
37