Subash-Khanal commited on
Commit
5f32be7
·
verified ·
1 Parent(s): a0573b3

quick-start SR uses showcase config (cfg 2.5 + bicubic warm start)

Browse files
Files changed (1) hide show
  1. README.md +6 -1
README.md CHANGED
@@ -89,9 +89,14 @@ dino = hf_hub_download("MVRL/genesis", "dinov3_vitl16_pretrain_sat493m-eadcf0
89
  tile = fetch_tile_at_zoom(lon=151.2153, lat=-33.8568, zoom=16) # one real 256×256 tile
90
  tile.save(f"{out}/01_original_tile.png")
91
 
92
- # SR: z16 parent → 512×512 mosaic of its four z17 children (50 steps, cfg 1.0)
 
 
93
  tile.save(f"{out}/02_sr_input.png")
94
  sr_model, _ = load_sr_denoiser(sr_ckpt, "JiT-H/16", dino_weights=dino, device=device)
 
 
 
95
  run_superresolution(sr_model, tile, device, target_zoom=17).save(f"{out}/03_sr_output.png")
96
 
97
  # OP: keep the upper-left quadrant real, outpaint the other three (white = hole)
 
89
  tile = fetch_tile_at_zoom(lon=151.2153, lat=-33.8568, zoom=16) # one real 256×256 tile
90
  tile.save(f"{out}/01_original_tile.png")
91
 
92
+ # SR: z16 parent → 512×512 mosaic of its four z17 children.
93
+ # Showcase config = the paper's pyramid-engine SR settings (cfg 2.5 + bicubic
94
+ # warm start); the engine reads `cfg_scale`, so set the attribute (not args.cfg).
95
  tile.save(f"{out}/02_sr_input.png")
96
  sr_model, _ = load_sr_denoiser(sr_ckpt, "JiT-H/16", dino_weights=dino, device=device)
97
+ sr_model.cfg_scale = 2.5
98
+ sr_model.args.bicubic_init = True
99
+ sr_model.args.bicubic_init_t_start = 0.4
100
  run_superresolution(sr_model, tile, device, target_zoom=17).save(f"{out}/03_sr_output.png")
101
 
102
  # OP: keep the upper-left quadrant real, outpaint the other three (white = hole)