Spaces:
Running
on
Zero
Running
on
Zero
Y Phung Nguyen
commited on
Commit
·
e52570b
1
Parent(s):
30bc748
Add Dt
Browse files
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"
|