openfree commited on
Commit
3e45a0e
·
verified ·
1 Parent(s): fca6281

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -5,7 +5,8 @@ os.environ["TORCH_DYNAMO_DISABLE"] = "1"
5
  import torch
6
  # 성능 최적화를 위한 설정 (TensorFloat32 연산 활성화)
7
  torch.set_float32_matmul_precision('high')
8
-
 
9
  import torch._dynamo
10
  import gradio as gr
11
  import spaces
 
5
  import torch
6
  # 성능 최적화를 위한 설정 (TensorFloat32 연산 활성화)
7
  torch.set_float32_matmul_precision('high')
8
+ import torch._inductor
9
+ torch._inductor.config.triton.cudagraphs = False
10
  import torch._dynamo
11
  import gradio as gr
12
  import spaces