Upload 2 files

slimshard_gui.py

@@ -48,6 +48,7 @@ from .slimshard import (
     HPCConfig,
     build_slimshard,
     export_hf_bundle,
+    export_safetensors_to_gguf,
     inspect_recovery_plan,
     recover_from_shard,
     regenerate,
@@ -267,6 +268,12 @@ class SlimShardGUI:
             text="Regenerate from SlimShard",
             variable=self.op_mode_var,
             value="regenerate",
+        ).pack(side=tk.LEFT, padx=(0, 14))
+        ttk.Radiobutton(
+            mode_bar,
+            text="Export SafeTensors to GGUF (Q2_K / Q4_0 / BF16)",
+            variable=self.op_mode_var,
+            value="export-gguf",
         ).pack(side=tk.LEFT)
 
         self._add_path_row(
@@ -335,11 +342,11 @@ class SlimShardGUI:
             state="readonly",
         ).pack(side=tk.LEFT, padx=(0, 16))
 
-        ttk.Label(row_opt, text="SlimShard Format:").pack(side=tk.LEFT, padx=(0, 6))
+        ttk.Label(row_opt, text="Format:").pack(side=tk.LEFT, padx=(0, 6))
         ttk.Combobox(
             row_opt,
             textvariable=self.quant_format_var,
-            values=("mixed", "q2_k", "q4_0", "exact"),
+            values=("mixed", "q2_k", "q4_0", "bf16", "exact"),
             width=9,
             state="readonly",
         ).pack(side=tk.LEFT, padx=(0, 16))
@@ -882,6 +889,23 @@ class SlimShardGUI:
                         stop_event=self._stop_event,
                     )
                     self._queue.put(("done", ("SlimShard Build Complete", res)))
+                elif mode == "export-gguf":
+                    shard = self.shard_var.get().strip()
+                    cfg = self.config_var.get().strip() or None
+                    fmt = self.quant_format_var.get()
+                    if fmt == "exact":
+                        fmt = "bf16"
+                    res = export_safetensors_to_gguf(
+                        shard,
+                        out,
+                        config=cfg,
+                        quant_format=fmt,
+                        device=self.device_var.get(),
+                        verbose=False,
+                        progress_cb=progress_cb,
+                        stop_event=self._stop_event,
+                    )
+                    self._queue.put(("done", ("GGUF Export Complete", res)))
                 else:
                     shard = self.shard_var.get().strip()
                     res = regenerate(