Closed vkbest closed 1 year ago
@vkbest I have the same problem with my M1 Max 32GB. Did you find a solution?
Traceback (most recent call last): File "/Users/username/kohya_ss/venv/bin/accelerate", line 8, in
sys.exit(main()) File "/Users/username/kohya_ss/venv/lib/python3.9/site-packages/accelerate/commands/accelerate_cli.py", line 45, in main args.func(args) File "/Users/username/kohya_ss/venv/lib/python3.9/site-packages/accelerate/commands/launch.py", line 1104, in launch_command simple_launcher(args) File "/Users/username/kohya_ss/venv/lib/python3.9/site-packages/accelerate/commands/launch.py", line 567, in simple_launcher raise subprocess.CalledProcessError(returncode=process.returncode, cmd=cmd) subprocess.CalledProcessError: Command '['/Users/username/kohya_ss/venv/bin/python', 'train_network.py', '--v2', '--enable_bucket', '--pretrained_model_name_or_path=/Users/username/stable-diffusion-webui/models/Stable-diffusion/v2-1_512-ema-pruned.safetensors', '--train_data_dir=/Users/username/stable-diffusion-webui/mymodels/mertk/lora/img', '--resolution=512,512', '--output_dir=/Users/username/stable-diffusion-webui/mymodels/mertk/lora/model', '--logging_dir=/Users/username/stable-diffusion-webui/mymodels/mertk/lora/log', '--network_alpha=128', '--save_model_as=safetensors', '--network_module=networks.lora', '--text_encoder_lr=5e-5', '--unet_lr=0.0001', '--network_dim=128', '--output_name=mertk', '--lr_scheduler_num_cycles=1', '--learning_rate=0.0001', '--lr_scheduler=constant', '--train_batch_size=2', '--max_train_steps=2900', '--save_every_n_epochs=1', '--mixed_precision=no', '--save_precision=float', '--seed=1234', '--caption_extension=.txt', '--cache_latents', '--optimizer_type=AdamW', '--max_data_loader_n_workers=1', '--clip_skip=2', '--bucket_reso_steps=64', '--xformers', '--bucket_no_upscale']' returned non-zero exit status 1.
I have configured the Mac and training but if I try to get preview every x steps or epochs I get this error:
Looks like the preview sample part try to force making a preview in CUDA, despite I'm using MPS (metal performance shaders from Apple) for training.