Spaces:
mikitona
/
Running on Zero

mikitona commited on
Commit
300728b
·
verified ·
1 Parent(s): 12a09c5

Update src/gradio_s3diff.py

Browse files
Files changed (1) hide show
  1. 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=60) # GPUを利用する関数にデコレーターを追加
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,