File size: 1,798 Bytes
144f05a | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 | {
"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
} |