fix(conversion): recognize "safetensor" anywhere in filename

kiri-art · May 25, 2023 · 1ceab7d · 1ceab7d
1 parent e3e13df
commit 1ceab7d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/api/convert_to_diffusers.py b/api/convert_to_diffusers.py
@@ -116,7 +116,7 @@ def main(
             "original_config_file": configPath if checkpoint_config_url else None,
             "device": device_id,
             "extract_ema": True,
-            "from_safetensors": fname.endswith(".safetensors"),
+            "from_safetensors": "safetensor" in fname.lower(),
         }
     )