Merge pull request cloneofsimo#175 from arceus-jia/master

fix "local variable 'unet_path' referenced before assignment"
tengshaofeng · Feb 9, 2023 · 6b82538 · 6b82538
2 parents 6f82996 + d7385a6
commit 6b82538
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)