Skip to content

Commit 9f80def

Browse files
committed
Fix noise level range check
Signed-off-by: ArchieMeng <archiemeng@protonmail.com>
1 parent ffd038d commit 9f80def

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

waifu2x_ncnn_vulkan_python/waifu2x_ncnn_vulkan.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,7 @@ def __init__(
5050

5151
# check arguments' validity
5252
assert scale in (1, 2), "scale must be 1 or 2"
53-
assert noise in range(4), "noise must be 0-4"
53+
assert noise in range(-1, 4), "noise must be [-1, 3]"
5454
assert isinstance(tta_mode, bool), "tta_mode isn't a boolean value"
5555
assert num_threads >= 1, "num_threads must be an integer >= 1"
5656

0 commit comments

Comments
 (0)