leffa/conf/model/cat_vton_flux.yaml

_target_: leffa.models.cat_vton_model.CatVtonModel
# FLUX.1-dev
pretrained_model_name_or_path: manifold://genads_models/tree/zijianzhou/model/FLUX.1-dev
new_in_channels: 33  # 16+1+16
height: ${constants.height}
width: ${constants.width}
garment_dropout_ratio: 0.1
use_dream: false
dream_detail_preservation: 10.0
use_garment_mask: false
only_optimize_unet_attn1: true
use_learning_flow_in_attention: false
learning_flow_in_attention_lambda: 0.001
learning_flow_in_attention_stop_timestep: 500
use_attention_flow_loss: false
attention_flow_loss_lambda: 0.001
use_pixel_space_supervision: false
pixel_space_supervision_lambda: 10.0
use_densepose: false