Y Phung Nguyen commited on
Commit
e52570b
·
1 Parent(s): 30bc748
Files changed (1) hide show
  1. config.py +2 -1
config.py CHANGED
@@ -6,7 +6,8 @@ import threading
6
  MEDSWIN_MODELS = {
7
  "MedSwin SFT": "MedSwin/MedSwin-7B-SFT",
8
  "MedSwin KD": "MedSwin/MedSwin-7B-KD",
9
- "MedSwin TA": "MedSwin/MedSwin-Merged-TA-SFT-0.7"
 
10
  }
11
  DEFAULT_MEDICAL_MODEL = "MedSwin TA"
12
  EMBEDDING_MODEL = "abhinand/MedEmbed-large-v0.1"
 
6
  MEDSWIN_MODELS = {
7
  "MedSwin SFT": "MedSwin/MedSwin-7B-SFT",
8
  "MedSwin KD": "MedSwin/MedSwin-7B-KD",
9
+ "MedSwin TA": "MedSwin/MedSwin-Merged-TA-SFT-0.7",
10
+ "MedSwin DT": "MedSwin/MedSwin-Merged-DaRE-TIES-KD-0.7",
11
  }
12
  DEFAULT_MEDICAL_MODEL = "MedSwin TA"
13
  EMBEDDING_MODEL = "abhinand/MedEmbed-large-v0.1"