ZhengPeng7
commited on
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
|