ensure FAST_SAFETENSORS_NUM_THREADS type is int (#91)

load safetensors support direct_io

rename env

Author: qzzz95

diffsynth_engine/utils/loader.py

@@ -18,7 +18,11 @@ def load_file(path: str | os.PathLike, device: str = "cpu"):
     if use_fast_safetensors:
         logger.info(f"FastSafetensors load model from {path}")
         start_time = time.time()
-        result = load_safetensors(str(path), num_threads=os.environ.get("FAST_SAFETENSORS_NUM_THREADS", 16))
+        result = load_safetensors(
+            str(path),
+            num_threads=int(os.environ.get("FAST_SAFETENSORS_NUM_THREADS", 16)),
+            direct_io=(os.environ.get("FAST_SAFETENSORS_DIRECT_IO", "False").upper() == "TRUE"),
+        )
         logger.info(f"FastSafetensors Load Model End. Time: {time.time() - start_time:.2f}s")
         return result
     else: