Update gradio_s3diff.py
Browse files- gradio_s3diff.py +1 -1
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=100) # GPUを利用する関数にデコレーターを追加 60 to 100
|
70 |
@torch.no_grad()
|
71 |
def process(
|
72 |
input_image: Image.Image,
|