paindiffusion / config.json
damtien440's picture
Push model using huggingface_hub.
e969e6c verified
{
"context_size": 2,
"ctrl_key": "ctrl",
"data_key": "x",
"drop_probs": [
0.1,
0.1,
0.1
],
"frame_stack": 4,
"guide": [
0.25,
0.5,
1.0
],
"img_size": 11,
"lognorm_mean": -1.2,
"lognorm_std": 1.2,
"loss_type": "L1",
"ode_solver": "dpm++",
"rho_schedule": 7,
"sample_output_dir": "/media/tien/SSD-NOT-OS/pain_intermediate_data/experiment/scale_jawpose_window32/sample",
"sample_steps": 35,
"self_cond": false,
"sigma_data": 0.5,
"sigma_max": 80,
"sigma_min": 0.002,
"time_delta": 0.1,
"uncertainty_scale": 2,
"warmup_steps": 1000,
"window_size": 8
}