change import from lora_keys to lora_keys_sd15

by321 · Jan 18, 2024 · 94a28ab · 94a28ab
1 parent 96c01ac
commit 94a28ab
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/safetensors_worker.py b/safetensors_worker.py
@@ -166,7 +166,7 @@ def ExtractHeader(cmdLine:dict,input_file:str,output_file:str)->int:
 
 
 def _CheckLoRA_internal(s:SafeTensorsFile)->int:
-    import lora_keys
+    import lora_keys_sd15 as lora_keys
     js=s.get_header()
     set_scalar=set()
     set_nonscalar=set()