| { |
| "model_name": "DI2FIX", |
| "model_repo_id": "ChengYou305/DI2FIX", |
| "checkpoint_filename": "model_80001.pkl", |
| "base_difix_repo_id": "nvidia/difix_ref", |
| "sd_turbo_repo_id": "stabilityai/sd-turbo", |
| "model_class": "Di2fix", |
| "pipeline_class": "DifixPipeline", |
| "lora_rank_vae": 4, |
| "timestep": 199, |
| "mv_unet": true, |
| "prompt": "remove degradation", |
| "max_side": 536, |
| "factor": 8, |
| "pad_to_square": true, |
| "weight_dtype": "float32", |
| "num_views": 2, |
| "source_view_index": 0, |
| "reference_view_index": 1, |
| "input_normalization": { |
| "mean": [ |
| 0.5 |
| ], |
| "std": [ |
| 0.5 |
| ] |
| }, |
| "output_format": "png" |
| } |
|
|
