MOSS-Image-IDM-2-Steps / training_args.json
akhaliq's picture
akhaliq HF Staff
TextBoost LoRA r64, 100 dense-text imgs, lr 1e-4
ebf10ad verified
Raw History Blame Contribute Delete
1.91 kB
{
"dataset_base_path": "/root/.cache/huggingface/hub/datasets--akhaliq--moss-textboost-data/snapshots/5464269627624e33cbc11f288d0145d5115e4491",
"dataset_metadata_path": "/root/.cache/huggingface/hub/datasets--akhaliq--moss-textboost-data/snapshots/5464269627624e33cbc11f288d0145d5115e4491/metadata.csv",
"dataset_repeat": 5,
"dataset_num_workers": 4,
"data_file_keys": "image,video",
"model_paths": null,
"model_id_with_origin_paths": "OpenMOSS-Team/MOSS-Image-SFT:transformer/diffusion_pytorch_model*.safetensors,OpenMOSS-Team/MOSS-Image-SFT:text_encoder/model*.safetensors,OpenMOSS-Team/MOSS-Image-SFT:vae/diffusion_pytorch_model.safetensors",
"extra_inputs": null,
"fp8_models": null,
"offload_models": null,
"resume_from_checkpoint": null,
"quant_options": null,
"learning_rate": 0.0001,
"num_epochs": 1,
"trainable_models": null,
"find_unused_parameters": true,
"weight_decay": 0.01,
"task": "sft",
"customized_optimizer": null,
"output_path": "/data/out",
"remove_prefix_in_ckpt": "pipe.dit.",
"save_steps": 250,
"lora_base_model": "dit",
"lora_target_modules": "",
"lora_rank": 64,
"lora_checkpoint": null,
"preset_lora_path": null,
"preset_lora_model": null,
"use_gradient_checkpointing": true,
"use_gradient_checkpointing_offload": false,
"gradient_accumulation_steps": 4,
"template_model_id_or_path": null,
"enable_lora_hot_loading": false,
"enable_model_cpu_offload": false,
"enable_optimizer_cpu_offload": false,
"cpu_offload_split_threshold": null,
"enable_tensorboard_log": false,
"enable_swanlab_log": false,
"swanlab_project": "DiffSynth-Studio",
"enable_wandb_log": false,
"wandb_project": "DiffSynth-Studio",
"enable_csv_log": true,
"height": 1024,
"width": 1024,
"max_pixels": 1048576,
"tokenizer_path": null
}