Skip to content

Commit 8fc694e

Browse files
Allow ESRGAN models to run with 1x and 8x scale
1 parent a0f3e1f commit 8fc694e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/visp/arch/esrgan.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -86,7 +86,7 @@ esrgan_params esrgan_detect_params(model_file const& f) {
8686
p.scale = f.get_int("esrgan.scale");
8787
p.n_blocks = f.get_int("esrgan.block_count");
8888

89-
if (p.scale < 2 || p.scale > 4) {
89+
if (p.scale < 1 || p.scale > 8) {
9090
throw except("ESRGAN: unsupported scale: {}", p.scale);
9191
}
9292
if (p.n_blocks < 1 || p.n_blocks > 23) {
@@ -100,4 +100,4 @@ int esrgan_estimate_graph_size(esrgan_params const& p) {
100100
return 512 + p.n_blocks * 192;
101101
}
102102

103-
} // namespace visp
103+
} // namespace visp

0 commit comments

Comments
 (0)