Charlie81
/

ThinExperts

Model card Files Files and versions

Charlie81 commited on Jun 9, 2025

Commit

4091143

·

1 Parent(s): dff49bf

args routers

Files changed (1) hide show

scripts/eval.py +1 -1

scripts/eval.py CHANGED Viewed

@@ -81,7 +81,7 @@ Examples:
         "--routing_type",
         type=str,
         default="topk",
-        choices=["topk", "multinomial", "botk", "topk+botk", "nth-descending"],
         help="Routing type (only used with custom models)"
     )
     parser.add_argument(

         "--routing_type",
         type=str,
         default="topk",
+        choices=["topk", "multinomial", "botk", "topk+botk", "nth-descending", "depthconstant", "depthlatter"],
         help="Routing type (only used with custom models)"
     )
     parser.add_argument(