Upload slimshard.py

slimshard.py

@@ -2297,6 +2297,8 @@ def main() -> None:
     rec.add_argument("--output", required=True, help="full model .safetensors")
     rec.add_argument("--config", default=None, help="config.json (defaults to index/shard dir)")
     rec.add_argument("--export-hf-dir", default=None, help="optional ready-to-load HF model directory")
+    rec.add_argument("--target-params-b", type=float, default=None, help="optional target parameter budget in billions (e.g. 30.0)")
+    rec.add_argument("--max-experts", type=int, default=None, help="optional max routed experts per layer to keep")
     rec.add_argument("--device", default="auto", choices=("auto", "cuda", "cpu"))
 
     args = ap.parse_args()
@@ -2325,6 +2327,8 @@ def main() -> None:
             device=args.device,
             verbose=True,
             export_hf_dir=args.export_hf_dir,
+            target_params_b=args.target_params_b,
+            max_experts=args.max_experts,
         )
     else:
         print(json.dumps(slimshard_info(args.input), indent=2))