Spaces:
Sleeping
Sleeping
Update main.py
#1
by
ricardo238costa
- opened
main.py
CHANGED
@@ -17,6 +17,7 @@ app.add_middleware(
|
|
17 |
allow_credentials=True,
|
18 |
)
|
19 |
|
|
|
20 |
device = "cuda:0" if torch.cuda.is_available() else "cpu"
|
21 |
torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
|
22 |
|
|
|
17 |
allow_credentials=True,
|
18 |
)
|
19 |
|
20 |
+
|
21 |
device = "cuda:0" if torch.cuda.is_available() else "cpu"
|
22 |
torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
|
23 |
|