diffusion/pipeline.json

{
    "name": "TrellisImageTo3DPipeline",
    "args": {
        "models": {
            "sparse_structure_decoder": "./pretrain/trellis/ckpts/ss_dec_conv3d_16l8_fp16",
            "sparse_structure_flow_model": "./pretrain/trellis/ckpts/ss_flow_img_dit_L_16l8_fp16",
            "slat_decoder_gs": "./pretrain/trellis/ckpts/slat_dec_gs_swin8_B_64l8gs32_fp16",
            "slat_decoder_rf": "./pretrain/trellis/ckpts/slat_dec_rf_swin8_B_64l8r16_fp16",

            "slat_decoder_mesh": "ckpts_new/decoder_step0100000.pt",
            "slat_decoder_phy": "ckpts_new/property_decoder_step0100000.pt",
            "slat_decoder_output": "ckpts_new/property_output_step0100000.pt",
            "slat_flow_model": "ckpts_new/denoiser_step0700000.pt",
            "slat_flow_model_phy": "ckpts_new/denoiser_phy_step0700000.pt"
        },
        "sparse_structure_sampler": {
            "name": "FlowEulerGuidanceIntervalSampler",
            "args": {
                "sigma_min": 1e-5
            },
            "params": {
                "steps": 25,
                "cfg_strength": 5.0,
                "cfg_interval": [0.5, 1.0],
                "rescale_t": 3.0
            }
        },
        "slat_sampler": {
            "name": "FlowEulerGuidanceIntervalSampler",
            "args": {
                "sigma_min": 1e-5
            },
            "params": {
                "steps": 25,
                "cfg_strength": 5.0,
                "cfg_interval": [0.5, 1.0],
                "rescale_t": 3.0
            }
        },
        "slat_normalization": {
            "mean": [
                -2.1687545776367188,
                -0.004347046371549368,
                -0.13352349400520325,
                -0.08418072760105133,
                -0.5271206498146057,
                0.7238689064979553,
                -1.1414450407028198,
                1.2039363384246826
            ],
            "std": [
                2.377650737762451,
                2.386378288269043,
                2.124418020248413,
                2.1748552322387695,
                2.663944721221924,
                2.371192216873169,
                2.6217446327209473,
                2.684523105621338
            ],
            "mean_phy": [
                -2.1507165, 
                -0.9456348, 
                -2.0234883, 
                -0.5949867, 
                -3.608296 ,
                -1.062877 , 
                -3.288852 , 
                -1.0749111
            ],
            "std_phy": [
                0.6931998 , 
                0.9221464 , 
                0.6542199 , 
                0.6594776 , 
                0.8451334 ,
                0.594917  , 
                0.69759405, 
                1.1614994 
            ]
        },
        "image_cond_model": "dinov2_vitl14_reg"
    }
}