Fixed scripts using installed versions of spandrel (#286)

chaiNNer-org · Jul 8, 2024 · ebf11ba · ebf11ba
1 parent 64d3c2b
commit ebf11ba
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/scripts/dump_state_dict.py b/scripts/dump_state_dict.py
@@ -43,13 +43,17 @@
 
 import argparse
 import collections
+import sys
 import textwrap
 from dataclasses import dataclass
 from typing import Any, Dict, Generic, Iterable, Mapping, TypeVar
 
 from torch import Tensor
 
 try:
+    sys.path.insert(0, __file__ + "/../../libs/spandrel")
+    sys.path.insert(0, __file__ + "/../../libs/spandrel_extra_arches")
+
     from spandrel import MAIN_REGISTRY, ModelLoader  # noqa: E402
     from spandrel_extra_arches import EXTRA_REGISTRY  # noqa: E402
 except ImportError: