Spaces:
Running
on
Zero
Running
on
Zero
ZhengPeng7
commited on
Commit
•
6be00d8
1
Parent(s):
006bbbb
Set presion of torch to high for consistency with training and faster inference.
Browse files
app.py
CHANGED
@@ -9,6 +9,7 @@ import gradio as gr
|
|
9 |
import spaces
|
10 |
from gradio_imageslider import ImageSlider
|
11 |
|
|
|
12 |
torch.jit.script = lambda f: f
|
13 |
|
14 |
from models.birefnet import BiRefNet
|
|
|
9 |
import spaces
|
10 |
from gradio_imageslider import ImageSlider
|
11 |
|
12 |
+
torch.set_float32_matmul_precision('high')
|
13 |
torch.jit.script = lambda f: f
|
14 |
|
15 |
from models.birefnet import BiRefNet
|