Spaces:
Running
Running
Sadjad Alikhani
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -26,10 +26,6 @@ if torch.cuda.is_available():
|
|
26 |
torch.backends.cudnn.deterministic = True
|
27 |
torch.backends.cudnn.benchmark = False
|
28 |
|
29 |
-
# Ensure that the model uses float32 precision (same as on GPU)
|
30 |
-
# Assuming you load your model later, ensure it's float precision.
|
31 |
-
# model = model.float() # Uncomment this when loading the model
|
32 |
-
|
33 |
# Enable deterministic algorithms in PyTorch (slower, but ensures consistency)
|
34 |
torch.use_deterministic_algorithms(True)
|
35 |
|
|
|
26 |
torch.backends.cudnn.deterministic = True
|
27 |
torch.backends.cudnn.benchmark = False
|
28 |
|
|
|
|
|
|
|
|
|
29 |
# Enable deterministic algorithms in PyTorch (slower, but ensures consistency)
|
30 |
torch.use_deterministic_algorithms(True)
|
31 |
|