pitt111 commited on
Commit
16bd7d5
·
verified ·
1 Parent(s): e35af71

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -67,8 +67,8 @@ from llm2vec import LLM2Vec
67
 
68
  # --- 1. Define Model IDs ---
69
  base_model_id = "google/codegemma-7b-it"
70
- mntp_model_id = "[SYSUSELab/DCS-CodeGemma-7b-It-MNTP]"
71
- supcon_model_id = "[SYSUSELab/DCS-CodeGemma-7b-It-SupCon-CSN]"
72
 
73
  # --- 2. Load Base Model and MNTP Adapter ---
74
  tokenizer = AutoTokenizer.from_pretrained(base_model_id)
 
67
 
68
  # --- 1. Define Model IDs ---
69
  base_model_id = "google/codegemma-7b-it"
70
+ mntp_model_id = "SYSUSELab/DCS-CodeGemma-7b-It-MNTP"
71
+ supcon_model_id = "SYSUSELab/DCS-CodeGemma-7b-It-SupCon-CSN"
72
 
73
  # --- 2. Load Base Model and MNTP Adapter ---
74
  tokenizer = AutoTokenizer.from_pretrained(base_model_id)