Update validation

bmaltais · Mar 1, 2024 · 89babc1 · 89babc1
1 parent a2450b1
commit 89babc1
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/requirements.txt b/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

diff --git a/setup/validate_requirements.py b/setup/validate_requirements.py
@@ -16,7 +16,7 @@
 # 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()

diff --git a/tools/lycoris_locon_extract.py b/tools/lycoris_locon_extract.py
@@ -134,8 +134,8 @@ def get_args():
 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()