Fix for LyCoris extraction

bmaltais · Aug 19, 2023 · 49c67e5 · 49c67e5
1 parent aafd42c
commit 49c67e5
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -10,10 +10,10 @@ fairscale==0.4.13
 ftfy==6.1.1
 gradio==3.36.1
 huggingface-hub==0.15.1
-lion-pytorch==0.0.6
-lycoris_lora==1.8.3
 # for loading Diffusers' SDXL
 invisible-watermark==0.2.0
+lion-pytorch==0.0.6
+lycoris_lora==1.8.3
 # open clip for SDXL
 open-clip-torch==2.20.0
 opencv-python==4.7.0.68

diff --git a/tools/lycoris_locon_extract.py b/tools/lycoris_locon_extract.py
@@ -86,7 +86,7 @@ def get_args():
 
 
 from lycoris.utils import extract_diff
-from lycoris.kohya_model_utils import load_models_from_stable_diffusion_checkpoint
+from lycoris.kohya.model_utils import load_models_from_stable_diffusion_checkpoint
 
 import torch
 from safetensors.torch import save_file