aljaz/InvSR
1
0
Fork 0
mirror of https://github.com/aljazceru/InvSR.git synced 2025-12-17 06:14:22 +01:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #3 from TezRomacH/patch-1

Browse Source 
Make seed useful
This commit is contained in:
Zongsheng Yue
2024-12-18 10:51:37 +08:00
committed by GitHub
parent 64dc14fff7 4061d99518
commit 824d42d8a6
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
app.py
View File

@@ -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.model_start.ckpt_path = str(started_ckpt_path)
configs.bs = 1 configs.bs = 1
configs.seed = 12345 configs.seed = seed
configs.basesr.chopping.pch_size = chopping_size configs.basesr.chopping.pch_size = chopping_size
configs.basesr.chopping.extra_bs = 4 configs.basesr.chopping.extra_bs = 4
return configs return configs
def predict(in_path, num_steps=1, chopping_size=128, seed=12345): 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) sampler = InvSamplerSR(configs)