mirror of https://github.com/bmaltais/kohya_ss
Update lycoris_locon_extract.py, updated imports (#3184)
parent
b9028e8710
commit
71ebffaafb
|
|
@ -124,8 +124,8 @@ ARGS = get_args()
|
|||
|
||||
|
||||
from lycoris.utils import extract_diff
|
||||
from lycoris.kohya.model_utils import load_models_from_stable_diffusion_checkpoint
|
||||
from lycoris.kohya.sdxl_model_util import load_models_from_sdxl_checkpoint
|
||||
from library.model_util import load_models_from_stable_diffusion_checkpoint
|
||||
from library.sdxl_model_util import load_models_from_sdxl_checkpoint
|
||||
|
||||
import torch
|
||||
from safetensors.torch import save_file
|
||||
|
|
|
|||
Loading…
Reference in New Issue