accelerate launch --num_cpu_threads_per_process=2 "./train_textual_inversion.py" --enable_bucket --min_bucket_reso=256 --max_bucket_reso=2048 --pretrained_model_name_or_path="/home/malcolm/sd/stable-diffusion-webui/models/Stable-diffusion/v1-5-pruned-emaonly.safetensors" --train_data_dir="/home/malcolm/sd/embeddings/data/$1" --resolution="512,512" --output_dir="/home/malcolm/sd/embeddings/output/$1" --save_model_as=safetensors --vae="/home/malcolm/sd/stable-diffusion-webui/models/VAE/vae-ft-mse-840000-ema-pruned.ckpt" --output_name="$1-ti" --lr_scheduler_num_cycles="1" --max_data_loader_n_workers="0" --gradient_accumulation_steps=15 --no_half_vae --learning_rate="0.005" --lr_scheduler="constant" --train_batch_size="1" --max_train_steps="1000" --mixed_precision="bf16" --save_precision="bf16" --seed="1234" --cache_latents --optimizer_type="AdamW" --max_data_loader_n_workers="0" --bucket_reso_steps=64 --save_every_n_steps="200" --xformers --bucket_no_upscale --noise_offset=0.0 --token_string="$1" --init_word="woman" --num_vectors_per_token=5 --use_object_template

Edit
Pub: 26 Nov 2023 12:33 UTC
Views: 1559