fix k-diffusion

cli/train.py

@@ -298,7 +298,7 @@ def prepare_options():
         options.lora.in_json = os.path.join(args.process_dir, args.name + '.json')
     if args.type == 'lyco':
         log.info('train using lycoris network')
-        options.lora.output_dir = args.lyco_dir
+        options.lora.output_dir = args.lora_dir
         options.lora.network_module = 'lycoris.kohya'
         options.lora.in_json = os.path.join(args.process_dir, args.name + '.json')
     # lora specific

modules/sd_samplers_kdiffusion.py

@@ -15,7 +15,8 @@ from modules.script_callbacks import AfterCFGCallbackParams, cfg_after_cfg_callb
 # deal with k-diffusion imports
 k_sampling = None
 try:
-    import k_diffusion.sampling as k_sampling # pylint: disable=wrong-import-order
+    import k_diffusion # pylint: disable=wrong-import-order
+    k_sampling = k_diffusion.sampling
 except ImportError:
     pass
 try:

requirements.txt

@@ -59,7 +59,6 @@ pandas==1.5.3
 protobuf==3.20.3
 pytorch_lightning==1.9.4
 transformers==4.34.1
-tokenizers==0.14.1
 tomesd==0.1.3
 urllib3==1.26.15
 Pillow==9.5.0