config.json

{
  "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"
}