mikitona commited on
Commit
021bfcb
·
verified ·
1 Parent(s): 9c66e69

Update gradio_s3diff.py

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