Upload slimshard_gui.py

slimshard_gui.py

@@ -864,7 +864,34 @@ class SlimShardGUI:
                     cmd += ["--max-experts", str(me)]
             except Exception:
                 pass
+            if self.anchor_layers_var.get().strip():
+                cmd += ["--anchor-layers", self.anchor_layers_var.get().strip()]
             cmd += ["--device", self.device_var.get()]
+        elif mode == "fetch-hf":
+            if self.hf_url_var.get().strip():
+                cmd += ["--repo", self.hf_url_var.get().strip()]
+            if self.model_dir_var.get().strip():
+                cmd += ["--output-dir", self.model_dir_var.get().strip()]
+            try:
+                tp = float(self.target_params_b_var.get())
+                if tp > 0.0:
+                    cmd += ["--target-params-b", f"{tp:g}"]
+            except Exception:
+                pass
+            try:
+                me = int(self.max_experts_var.get())
+                if me > 0:
+                    cmd += ["--max-experts", str(me)]
+            except Exception:
+                pass
+            try:
+                na = int(self.num_anchor_layers_var.get())
+                cmd += ["--num-anchor-layers", str(na)]
+            except Exception:
+                pass
+            if self.anchor_layers_var.get().strip():
+                cmd += ["--anchor-layers", self.anchor_layers_var.get().strip()]
+            cmd += ["--mode", self.hf_fetch_mode_var.get()]
         elif mode == "build":
             if self.model_dir_var.get():
                 cmd += ["--model-dir", self.model_dir_var.get()]
@@ -891,8 +918,12 @@ class SlimShardGUI:
 
     def _inspect_plan(self) -> None:
         shard = self.shard_var.get().strip()
-        if not shard or not Path(shard).exists():
-            messagebox.showwarning("SlimShard", "Please select a valid .safetensors shard first.")
+        try:
+            resolved = _resolve_shard_paths(shard) if shard else []
+        except Exception:
+            resolved = []
+        if not resolved:
+            messagebox.showwarning("SlimShard", "Please select one or more valid .safetensors shards first.")
             return
         try:
             tp = float(self.target_params_b_var.get())
@@ -909,6 +940,7 @@ class SlimShardGUI:
                 config=self.config_var.get().strip() or None,
                 target_params_b=tp if tp > 0.0 else None,
                 max_experts=me if me > 0 else None,
+                anchor_layers=self.anchor_layers_var.get().strip() or None,
             )
         except Exception as exc:
             messagebox.showerror("Inspect Failed", str(exc))
@@ -929,7 +961,7 @@ class SlimShardGUI:
         surv = plan["surviving_layers"]
         miss = plan["missing_layers"]
         self.layers_summary_var.set(
-            f"Surviving in shard ({len(surv)}): {surv[:12]}{'...' if len(surv) > 12 else ''}  |  "
+            f"Surviving across {len(resolved)} shard(s) ({len(surv)}): {surv[:12]}{'...' if len(surv) > 12 else ''}  |  "
             f"Missing to CheckFree-reconstruct ({len(miss)}): {miss[:12]}{'...' if len(miss) > 12 else ''}"
         )
         c = plan["counts"]
@@ -940,8 +972,8 @@ class SlimShardGUI:
         self._plan_rows = plan["rows"]
         self._refresh_table()
         self._log(
-            f"Inspected {Path(shard).name}: {plan['present_tensors']}/{plan['total_tensors']} present, "
-            f"{len(miss)} layers to reconstruct via CheckFree."
+            f"Inspected {len(resolved)} shard(s): {plan['present_tensors']}/{plan['total_tensors']} present, "
+            f"{len(surv)} anchor layers, {len(miss)} layers to reconstruct via CheckFree."
         )
 
     def _refresh_table(self) -> None: