Spaces:
Running
on
Zero
Running
on
Zero
Fix the bug of random seed.
Browse files
app.py
CHANGED
|
@@ -52,7 +52,7 @@ def get_configs(num_steps=1, chopping_size=128, seed=12345):
|
|
| 52 |
configs.model_start.ckpt_path = str(started_ckpt_path)
|
| 53 |
|
| 54 |
configs.bs = 1
|
| 55 |
-
configs.seed =
|
| 56 |
configs.basesr.chopping.pch_size = chopping_size
|
| 57 |
if chopping_size == 128:
|
| 58 |
configs.basesr.chopping.extra_bs = 8
|
|
@@ -65,7 +65,7 @@ def get_configs(num_steps=1, chopping_size=128, seed=12345):
|
|
| 65 |
|
| 66 |
@spaces.GPU
|
| 67 |
def predict(in_path, num_steps=1, chopping_size=128, seed=12345):
|
| 68 |
-
configs = get_configs(num_steps=num_steps, chopping_size=chopping_size, seed=
|
| 69 |
|
| 70 |
sampler = InvSamplerSR(configs)
|
| 71 |
|
|
@@ -143,5 +143,5 @@ demo = gr.Interface(
|
|
| 143 |
)
|
| 144 |
|
| 145 |
demo.queue(max_size=5)
|
| 146 |
-
demo.launch(share=
|
| 147 |
|
|
|
|
| 52 |
configs.model_start.ckpt_path = str(started_ckpt_path)
|
| 53 |
|
| 54 |
configs.bs = 1
|
| 55 |
+
configs.seed = seed
|
| 56 |
configs.basesr.chopping.pch_size = chopping_size
|
| 57 |
if chopping_size == 128:
|
| 58 |
configs.basesr.chopping.extra_bs = 8
|
|
|
|
| 65 |
|
| 66 |
@spaces.GPU
|
| 67 |
def predict(in_path, num_steps=1, chopping_size=128, seed=12345):
|
| 68 |
+
configs = get_configs(num_steps=num_steps, chopping_size=chopping_size, seed=seed)
|
| 69 |
|
| 70 |
sampler = InvSamplerSR(configs)
|
| 71 |
|
|
|
|
| 143 |
)
|
| 144 |
|
| 145 |
demo.queue(max_size=5)
|
| 146 |
+
demo.launch(share=False)
|
| 147 |
|