pminervini commited on
Commit
7e68bad
1 Parent(s): afd391f
Files changed (1) hide show
  1. src/backend/envs.py +1 -1
src/backend/envs.py CHANGED
@@ -29,6 +29,6 @@ class Tasks(Enum):
29
  EVAL_REQUESTS_PATH_BACKEND = os.path.join(CACHE_PATH, "eval-queue-bk")
30
  EVAL_RESULTS_PATH_BACKEND = os.path.join(CACHE_PATH, "eval-results-bk")
31
 
32
- DEVICE = "cuda:0" if torch.cuda.is_available() else 'cpu'
33
 
34
  LIMIT = None # Testing; needs to be None
 
29
  EVAL_REQUESTS_PATH_BACKEND = os.path.join(CACHE_PATH, "eval-queue-bk")
30
  EVAL_RESULTS_PATH_BACKEND = os.path.join(CACHE_PATH, "eval-results-bk")
31
 
32
+ DEVICE = "cuda" if torch.cuda.is_available() else 'cpu'
33
 
34
  LIMIT = None # Testing; needs to be None