devilismyfriend / StableTuner

Finetuning SD in style.
GNU Affero General Public License v3.0
666 stars 51 forks source link

Using "Fine-Tune" settings causes division by zero error #97

Open NoteToSelfFindGoodNickname opened 1 year ago

NoteToSelfFindGoodNickname commented 1 year ago

Loading Latent Cache from models\iconex\logs\latent_cache Latents are ready. Traceback (most recent call last): File "C:\Users\tomwe\st4\scripts\trainer.py", line 2902, in main() File "C:\Users\tomwe\st4\scripts\trainer.py", line 2216, in main args.num_train_epochs = math.ceil(args.max_train_steps / num_update_steps_per_epoch) ZeroDivisionError: division by zero Traceback (most recent call last): File "C:\Users\tomwe\miniconda3\envs\ST\lib\runpy.py", line 196, in _run_module_as_main return _run_code(code, main_globals, None, File "C:\Users\tomwe\miniconda3\envs\ST\lib\runpy.py", line 86, in _run_code exec(code, run_globals) File "C:\Users\tomwe\miniconda3\envs\ST\Scripts\accelerate.exe__main__.py", line 7, in File "C:\Users\tomwe\miniconda3\envs\ST\lib\site-packages\accelerate\commands\accelerate_cli.py", line 45, in main args.func(args) File "C:\Users\tomwe\miniconda3\envs\ST\lib\site-packages\accelerate\commands\launch.py", line 1104, in launch_command simple_launcher(args) File "C:\Users\tomwe\miniconda3\envs\ST\lib\site-packages\accelerate\commands\launch.py", line 567, in simple_launcher raise subprocess.CalledProcessError(returncode=process.returncode, cmd=cmd) subprocess.CalledProcessError: Command '['C:\Users\tomwe\miniconda3\envs\ST\python.exe', 'scripts/trainer.py', '--sample_from_batch=2', '--attention=xformers', '--model_variant=base', '--normalize_masked_area_loss', '--unmasked_probability=0.0', '--max_denoising_strength=1.0', '--disable_cudnn_benchmark', '--use_text_files_as_captions', '--sample_step_interval=1', '--pretrained_model_name_or_path=C:\Users\tomwe\st4\models\galaxytimemachinesGTM_v1', '--pretrained_vae_name_or_path=', '--output_dir=models/iconex', '--seed=3434554', '--resolution=512', '--train_batch_size=41', '--num_train_epochs=100', '--mixed_precision=fp16', '--use_bucketing', '--aspect_mode=dynamic', '--aspect_mode_action_preference=add', '--use_8bit_adam', '--gradient_checkpointing', '--gradient_accumulation_steps=1', '--learning_rate=3e-6', '--lr_warmup_steps=0', '--lr_scheduler=constant', '--train_text_encoder', '--concepts_list=stabletune_concept_list.json', '--num_class_images=200', '--save_every_n_epoch=5', '--n_save_sample=2', '--sample_height=512', '--sample_width=512', '--dataset_repeats=1', '--add_sample_prompt=a banana by iconex', '--sample_on_training_start']' returned non-zero exit status 1.


However, when I click "Dreambooth", it works fine.