From 4061d995187a81412721e8fec7e962dc1b8d6245 Mon Sep 17 00:00:00 2001 From: Roman Tezikov Date: Tue, 17 Dec 2024 18:05:18 +0200 Subject: [PATCH] Make seed useful --- app.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/app.py b/app.py index 40fa13c..00d3c6f 100644 --- a/app.py +++ b/app.py @@ -56,14 +56,14 @@ def get_configs(num_steps=1, chopping_size=128, seed=12345): configs.model_start.ckpt_path = str(started_ckpt_path) configs.bs = 1 - configs.seed = 12345 + configs.seed = seed configs.basesr.chopping.pch_size = chopping_size configs.basesr.chopping.extra_bs = 4 return configs def predict(in_path, num_steps=1, chopping_size=128, seed=12345): - configs = get_configs(num_steps=num_steps, chopping_size=chopping_size, seed=12345) + configs = get_configs(num_steps=num_steps, chopping_size=chopping_size, seed=seed) sampler = InvSamplerSR(configs)