caption_prefix = '@riot.'# trained using diffusion-pipe commit 6e95020cad0b3cd7dcb93ce42b358669051bf6d2output_dir = '/home/txiaou/diffusion-pipe/output/riot_style'dataset = '/home/txiaou/diffusion-pipe/examples/riot.toml'eval_datasets = [{name = 'eval_train', config = '/home/txiaou/diffusion-pipe/examples/riot_eval_train.toml'},]# training settingsepochs = 80micro_batch_size_per_gpu = 1pipeline_stages = 1gradient_accumulation_steps = 4gradient_clipping = 1warmup_steps = 100# eval settingseval_every_n_epochs = 5#eval_every_n_steps = 500#eval_every_n_examples = 2048000eval_before_first_step = trueeval_micro_batch_size_per_gpu = 1eval_gradient_accumulation_steps = 1# misc settingssave_every_n_epochs = 10#save_every_n_steps = 1000#save_every_n_examples = 4096000#checkpoint_every_n_epochs = 1#checkpoint_every_n_minutes = 120activation_checkpointing = true#reentrant_activation_checkpointing = truepartition_method = 'parameters'# partition_method = 'manual'# partition_split = [10]save_dtype = 'bfloat16'caching_batch_size = 1map_num_proc = 8steps_per_print = 1compile = true[model]type = 'anima'transformer_path = '/home/txiaou/diffusion-pipe/models/anima/anima-base-v1.0.safetensors'vae_path = '/home/txiaou/diffusion-pipe/models/qwen/qwen_image_vae.safetensors'llm_path = '/home/txiaou/diffusion-pipe/models/qwen/qwen_3_06b_base.safetensors'dtype = 'bfloat16'#cache_text_embeddings = falsellm_adapter_lr = 0#timestep_sample_method = 'uniform'#flux_shift = true#multiscale_loss_weight = 0.5sigmoid_scale = 1.3[adapter]type = 'lora'rank = 32dtype = 'bfloat16'[optimizer]type = 'adamw_optimi'lr = 2e-5betas = [0.9, 0.99]weight_decay = 0.01eps = 1e-8

Files with SHA256: 8db7cbcb2203116c98f53632778c66e274beacbe4652a25e1a7caf02a376209b