Spaces:
Running
on
Zero
Running
on
Zero
yadonglu
commited on
Commit
·
ef66abf
1
Parent(s):
8b5dff8
app.py
CHANGED
|
@@ -42,7 +42,7 @@ OmniParser is a screen parsing tool to convert general GUI screen to structured
|
|
| 42 |
DEVICE = torch.device('cuda')
|
| 43 |
|
| 44 |
@spaces.GPU
|
| 45 |
-
|
| 46 |
# @torch.autocast(device_type="cuda", dtype=torch.bfloat16)
|
| 47 |
def process(
|
| 48 |
image_input,
|
|
|
|
| 42 |
DEVICE = torch.device('cuda')
|
| 43 |
|
| 44 |
@spaces.GPU
|
| 45 |
+
@torch.inference_mode()
|
| 46 |
# @torch.autocast(device_type="cuda", dtype=torch.bfloat16)
|
| 47 |
def process(
|
| 48 |
image_input,
|