Revert determinism

7dc1b8b3 · novelailab · a7c71667 · 7dc1b8b3 · 7dc1b8b3
Commit 7dc1b8b3 authored Sep 30, 2022 by novelailab
Hide whitespace changes
Inline Side-by-side

Showing with 0 additions and 7 deletions

hydra_node/config.py hydra_node/config.py +0 -4

main.py main.py +0 -3

No files found.
--- a/hydra_node/config.py
+++ b/hydra_node/config.py
 import os
-os.environ["CUBLAS_WORKSPACE_CONFIG"]=":4096:8"
-
 import torch
-torch.use_deterministic_algorithms(True, warn_only=True)
-
 import logging
 import os
 import platform

--- a/main.py
+++ b/main.py
-import os
-os.environ["CUBLAS_WORKSPACE_CONFIG"]=":4096:8"
-
 from fastapi import FastAPI, Request, Depends
 from pydantic import BaseModel
 from fastapi.responses import HTMLResponse, PlainTextResponse, Response