Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -8,6 +8,8 @@ import json
|
|
8 |
import spaces
|
9 |
import torch
|
10 |
|
|
|
|
|
11 |
torch.use_deterministic_algorithms(True)
|
12 |
torch.backends.cudnn.deterministic = True
|
13 |
torch.backends.cudnn.benchmark = False
|
|
|
8 |
import spaces
|
9 |
import torch
|
10 |
|
11 |
+
torch.set_num_threads(1)
|
12 |
+
torch.set_num_interop_threads(1)
|
13 |
torch.use_deterministic_algorithms(True)
|
14 |
torch.backends.cudnn.deterministic = True
|
15 |
torch.backends.cudnn.benchmark = False
|