fix "local variable 'unet_path' referenced before assignment" in patc…

…h_pipe
tengshaofeng · Feb 9, 2023 · d7385a6 · d7385a6
1 parent 848db91
commit d7385a6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/lora_diffusion/lora.py b/lora_diffusion/lora.py
@@ -974,6 +974,8 @@ def patch_pipe(
             unet_path = maybe_unet_path[:-6] + ".pt"
         elif maybe_unet_path.endswith(".text_encoder.pt"):
             unet_path = maybe_unet_path[:-16] + ".pt"
+        else:
+            unet_path = maybe_unet_path
 
         ti_path = _ti_lora_path(unet_path)
         text_path = _text_lora_path(unet_path)