devilismyfriend / StableTuner

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

Getting "sample_on_training_start']' returned non-zero exit status 1" #92

Open kyle215ps3 opened 1 year ago

kyle215ps3 commented 1 year ago

Traceback (most recent call last): File "C:\Users\123ky.conda\envs\ST\lib\runpy.py", line 196, in _run_module_as_main return _run_code(code, main_globals, None, File "C:\Users\123ky.conda\envs\ST\lib\runpy.py", line 86, in _run_code exec(code, run_globals) File "C:\Users\123ky.conda\envs\ST\Scripts\accelerate.exe__main__.py", line 7, in File "C:\Users\123ky.conda\envs\ST\lib\site-packages\accelerate\commands\accelerate_cli.py", line 45, in main args.func(args) File "C:\Users\123ky.conda\envs\ST\lib\site-packages\accelerate\commands\launch.py", line 1104, in launch_command simple_launcher(args) File "C:\Users\123ky.conda\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\123ky\.conda\envs\ST\python.exe', 'scripts/trainer.py', '--attention=xformers', '--model_variant=inpainting', '--normalize_masked_area_loss', '--unmasked_probability=0.0', '--max_denoising_strength=1.0', '--disable_cudnn_benchmark', '--use_text_files_as_captions', '--sample_step_interval=50', '--stop_text_encoder_training=15', '--pretrained_model_name_or_path=runwayml/stable-diffusion-inpainting', '--pretrained_vae_name_or_path=', '--output_dir=models/new_model', '--seed=3434554', '--resolution=512', '--train_batch_size=5', '--num_train_epochs=50', '--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', '--regenerate_latent_cache', '--train_text_encoder', '--use_image_names_as_captions', '--concepts_list=stabletune_concept_list.json', '--num_class_images=200', '--save_every_n_epoch=15', '--n_save_sample=1', '--sample_height=512', '--sample_width=512', '--dataset_repeats=2', '--add_sample_prompt=Alex Grey Art', '--sample_on_training_start']' returned non-zero exit status 1. Press any key to continue . . . Any idea what im doing wrong? I have a 3070 8GB vram also im overclocking as well