| model: | |
| __object__: | |
| path: src.models.pixl_relight | |
| name: PIXLRelight | |
| args: as_params | |
| net: | |
| __object__: | |
| path: src.models.net | |
| name: Net | |
| args: as_params | |
| img_size: 512 | |
| patch_size: 16 | |
| pos_embed_rope_base: 100.0 | |
| pos_embed_rope_min_period: null | |
| pos_embed_rope_max_period: null | |
| pos_embed_rope_normalize_coords: separate | |
| pos_embed_rope_shift_coords: null | |
| pos_embed_rope_jitter_coords: null | |
| pos_embed_rope_rescale_coords: null | |
| pos_embed_rope_dtype: bf16 | |
| embed_dim: 1024 | |
| depth: 24 | |
| num_heads: 16 | |
| ffn_ratio: 4.0 | |
| qkv_bias: True | |
| drop_path_rate: 0.0 | |
| layerscale_init: 1.0 | |
| norm_layer: layernormbf16 | |
| n_storage_tokens: 8 | |
| mask_k_bias: False | |
| qk_norm: True | |
| rgb_patch_embed: vit_large | |
| intrinsics_patch_embed: convnext_base | |
| intermediate_layer_idx: [4, 11, 17, 23] | |
| gradient_checkpointing: False | |
| head: | |
| __object__: | |
| path: src.models.head | |
| name: DPTHead | |
| args: as_params | |
| dim_in: 1024 | |
| patch_size: 16 | |
| output_dim: 6 | |
| activation: sigmoid | |
| conf_activation: null | |
| features: 256 | |
| out_channels: [256, 512, 1024, 1024] | |
| pos_embed: True | |
| head_name: rgb | |
| output_mode: modulation | |
| modulation_clamp: True | |
| rgbx: | |
| __object__: | |
| path: src.rgbx.inference | |
| name: Marigold | |
| args: as_params | |
| checkpoint_path: prs-eth/marigold-iid-lighting-v1-1 | |
| denoise_steps: 1 | |
| processing_res: 0 | |
| half_precision: True | |
| output_processing_res: True | |
| resample_method: bilinear |