Sarthak commited on
Commit ·
fba41e9
1
Parent(s): d8a91f9
feat(analyze): enhance model type identification in analysis
Browse filesThe logic has been updated to differentiate between simplified distilled models and fine-tuned distilled models, and assign a new model type for fine-tuned variations.
- src/distiller/analyze.py +5 -1
src/distiller/analyze.py
CHANGED
|
@@ -1699,7 +1699,11 @@ Our distilled models exhibit consistent architectural characteristics across dif
|
|
| 1699 |
or "distilled" in model_name_display.lower()
|
| 1700 |
or "(ours)" in model_name_display.lower()
|
| 1701 |
):
|
| 1702 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1703 |
elif any(code_term in model_name_display.lower() for code_term in ["codebert", "graphcode", "codet5"]):
|
| 1704 |
model_type = "Code-Specific"
|
| 1705 |
elif "potion" in model_name_display.lower():
|
|
|
|
| 1699 |
or "distilled" in model_name_display.lower()
|
| 1700 |
or "(ours)" in model_name_display.lower()
|
| 1701 |
):
|
| 1702 |
+
# Check if it's a fine-tuned model
|
| 1703 |
+
if "fine_tuned" in model_name_display.lower():
|
| 1704 |
+
model_type = "**🎓 Fine-tuned Distillation**"
|
| 1705 |
+
else:
|
| 1706 |
+
model_type = "**🔥 Simplified Distillation**"
|
| 1707 |
elif any(code_term in model_name_display.lower() for code_term in ["codebert", "graphcode", "codet5"]):
|
| 1708 |
model_type = "Code-Specific"
|
| 1709 |
elif "potion" in model_name_display.lower():
|