Slimshard

Allow execution from inside repo folder

slimshard.py

@@ -39,11 +39,16 @@ import os
 import re
 import shutil
 import struct
+import sys
 import tempfile
 from pathlib import Path
 
 import numpy as np
 
+if __package__ in (None, ""):
+    sys.path.insert(0, str(Path(__file__).resolve().parent.parent))
+    __package__ = "hpc_quantize_v2"
+
 try:
     import torch
     import torch.nn.functional as F

slimshard_gui.py

@@ -30,6 +30,7 @@ from __future__ import annotations
 import datetime as dt
 import queue
 import shlex
+import sys
 import threading
 import time
 import tkinter as tk
@@ -38,6 +39,10 @@ from tkinter import filedialog, messagebox, ttk
 
 import numpy as np
 
+if __package__ in (None, ""):
+    sys.path.insert(0, str(Path(__file__).resolve().parent.parent))
+    __package__ = "hpc_quantize_v2"
+
 from .cuda_backend import cuda_device_name, cuda_is_available
 from .slimshard import (
     HPCConfig,
@@ -569,14 +574,16 @@ class SlimShardGUI:
         )
 
     def _auto_populate_defaults(self) -> None:
-        cwd = Path.cwd()
-        raon_shard = cwd / "Raon-1-mini-preview" / "model-00001-of-00007.safetensors"
-        if raon_shard.exists():
-            self._set_shard_and_companions(raon_shard)
-        else:
-            shards = sorted(cwd.glob("*/*.safetensors"))
+        repo_root = Path(__file__).resolve().parent.parent
+        for base in (Path.cwd(), repo_root):
+            raon_shard = base / "Raon-1-mini-preview" / "model-00001-of-00007.safetensors"
+            if raon_shard.exists():
+                self._set_shard_and_companions(raon_shard)
+                return
+            shards = sorted(base.glob("*/*.safetensors"))
             if shards:
                 self._set_shard_and_companions(shards[0])
+                return
 
     def _set_shard_and_companions(self, shard_path: Path) -> None:
         self.shard_var.set(str(shard_path))