animax / train-params_animax-pure-ft_v01.txt
rityak's picture
Upload 2 files
577c7d8 verified
accelerate launch --num_cpu_threads_per_process 1 anima_train.py `
--pretrained_model_name_or_path="D:\Models\diffusion_models\anima-preview.safetensors" `
--qwen3="D:\Models\text_encoders\qwen_3_06b_base.safetensors" `
--vae="D:\Models\vae\qwen_image_vae.safetensors" `
--dataset_config="E:\Train\dataset.toml" `
--output_dir="E:\Train\Output" `
--output_name="anima_ft" `
--save_model_as=safetensors `
--max_train_epochs=3 `
--save_every_n_epochs=1 `
--mixed_precision="bf16" `
--optimizer_type="AdamW8bit" `
--learning_rate=5e-6 `
--lr_scheduler="cosine_with_restarts" `
--lr_warmup_steps=100 `
--max_grad_norm=1.0 `
--timestep_sampling="sigmoid" `
--discrete_flow_shift=1.0 `
--gradient_checkpointing `
--cache_latents `
--cache_text_encoder_outputs `
--vae_chunk_size=64 `
--vae_disable_cache `
--attn_mode="sdpa" `
--persistent_data_loader_workers `
--max_data_loader_n_workers=2 `
--logging_dir="E:\Train\Logs" `
--unsloth_offload_checkpointing `
--llm_adapter_lr 0 `
--full_bf16