| { |
| "dataset_base_path": "/content/mtrini_imagine_fast", |
| "dataset_metadata_path": "/content/mtrini_imagine_fast/metadata.csv", |
| "dataset_repeat": 1, |
| "dataset_num_workers": 8, |
| "data_file_keys": "image,video", |
| "model_paths": null, |
| "model_id_with_origin_paths": "HiDream-ai/HiDream-O1-Image: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": false, |
| "weight_decay": 0.01, |
| "task": "sft", |
| "customized_optimizer": null, |
| "output_path": "/content/mtrini_imagine_fast/output", |
| "remove_prefix_in_ckpt": "pipe.dit.", |
| "save_steps": null, |
| "lora_base_model": "dit", |
| "lora_target_modules": "q_proj,k_proj,v_proj,o_proj,gate_proj,up_proj,down_proj,attn.qkv,attn.proj,mlp.linear_fc1,mlp.linear_fc2", |
| "lora_rank": 16, |
| "lora_checkpoint": null, |
| "preset_lora_path": null, |
| "preset_lora_model": null, |
| "use_gradient_checkpointing": true, |
| "use_gradient_checkpointing_offload": false, |
| "gradient_accumulation_steps": 1, |
| "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": false, |
| "height": null, |
| "width": null, |
| "max_pixels": 262144, |
| "processor_config": "HiDream-ai/HiDream-O1-Image:./", |
| "noise_scale": 8.0, |
| "initialize_model_on_cpu": false |
| } |
