Update validation

requirements.txt

@@ -13,7 +13,7 @@ huggingface-hub==0.20.1
 # for loading Diffusers' SDXL
 invisible-watermark==0.2.0
 lion-pytorch==0.0.6
-lycoris_lora==2.0.2
+lycoris_lora==2.1.0.post3
 # for BLIP captioning
 # requests==2.28.2
 # timm==0.6.12

setup/validate_requirements.py

@@ -16,7 +16,7 @@ if "setup" in project_directory:
 # Add the project directory to the beginning of the Python search path
 sys.path.insert(0, project_directory)
 
-from library.custom_logging import setup_logging
+from kohya_gui.custom_logging import setup_logging
 
 # Set up logging
 log = setup_logging()

tools/lycoris_locon_extract.py

@@ -134,8 +134,8 @@ from safetensors.torch import save_file
 def main():
     args = ARGS
     if args.is_sdxl:
-        base = load_models_from_sdxl_checkpoint(None, args.base_model, args.device)
-        db = load_models_from_sdxl_checkpoint(None, args.db_model, args.device)
+        base = load_models_from_sdxl_checkpoint(None, args.base_model, "cpu")
+        db = load_models_from_sdxl_checkpoint(None, args.db_model, "cpu")
     else:
         base = load_models_from_stable_diffusion_checkpoint(args.is_v2, args.base_model)
         db = load_models_from_stable_diffusion_checkpoint(args.is_v2, args.db_model)
@@ -187,4 +187,4 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
+    main()