Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-20 01:31:13 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix training

Browse Source
This commit is contained in:
Vladimir Mandic
2023-07-04 15:40:15 -04:00
parent 191da73d48
commit 069c2ddb77
2 changed files with 20 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
cli/train.py
+8 -3
View File
@@ -265,6 +265,7 @@ def prepare_options():
options.lora.in_json = None
if args.type == 'dreambooth':
log.info('train using dreambooth style training')
options.lora.vae_batch_size = args.batch
options.lora.in_json = None
if args.type == 'lora':
log.info('train using lora style training')
@@ -378,14 +379,18 @@ def check_versions():
log.info('experimental mode enabled')
return
log.info('checking accelerate')
error = False
import accelerate
if accelerate.__version__ != '0.19.0':
log.error(f'invalid accelerate version: required=0.19.0 found={accelerate.__version__}')
exit(1)
log.error(f'invalid accelerate version: accelerate=0.19.0 found={accelerate.__version__}')
error = True
log.info('checking diffusers')
import diffusers
if diffusers.__version__ != '0.10.2':
log.error(f'invalid diffusers version: required=0.10.2 found={diffusers.__version__}')
log.error(f'invalid diffusers version: diffusers=0.10.2 found={diffusers.__version__}')
error = True
if error:
log.info('> pip install accelerate==0.19.0 diffusers==0.10.2')
exit(1)