Update src/gradio_s3diff.py
Browse files- src/gradio_s3diff.py +1 -1
src/gradio_s3diff.py
CHANGED
@@ -66,7 +66,7 @@ device = "cuda"
|
|
66 |
net_sr.to(device, dtype=weight_dtype)
|
67 |
net_de.to(device, dtype=weight_dtype)
|
68 |
|
69 |
-
@GPU(duration=
|
70 |
@torch.no_grad()
|
71 |
def process(
|
72 |
input_image: Image.Image,
|
|
|
66 |
net_sr.to(device, dtype=weight_dtype)
|
67 |
net_de.to(device, dtype=weight_dtype)
|
68 |
|
69 |
+
@GPU(duration=55) # GPUを利用する関数にデコレーターを追加
|
70 |
@torch.no_grad()
|
71 |
def process(
|
72 |
input_image: Image.Image,
|