Sarthak commited on
Commit
fba41e9
·
1 Parent(s): d8a91f9

feat(analyze): enhance model type identification in analysis

Browse files

The 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.

Files changed (1) hide show
  1. 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
- model_type = "**🔥 Simplified Distillation**"
 
 
 
 
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():