inference_config.json

{
  "repo_id": "W8Yi/distilled-wsi-diffusion",
  "source_ckpt": "/common/users/wq50/SAE_path/runs/pixcell_student_traj/ckpt/final.pt",
  "weights_key": "ema",
  "export_dtype": "fp16",
  "student_arch": "pixcell",
  "cond_dim": 1536,
  "latent_size": 32,
  "teacher_steps": 28,
  "guidance_student": 1.0,
  "default_sample_steps": 4,
  "pix_model_id": "StonyBrook-CVLab/PixCell-256",
  "pix_pipeline_id": "StonyBrook-CVLab/PixCell-pipeline",
  "vae_model_id": "stabilityai/stable-diffusion-3.5-large",
  "vae_subfolder": "vae"
}