Hugging Face's logo

Spaces:
sparse-generative-ai
/
open-moe-llm-leaderboard
Running

App Files Community
33
LuZhan commited on
Commit
da43b75
•
1 Parent(s): 2acbfac

fix:change cuda explicitly to cuda:0

Browse files
Files changed (1) hide show
  1. src/backend/envs.py +1 -1
src/backend/envs.py CHANGED
@@ -63,4 +63,4 @@ EVAL_REQUESTS_PATH_BACKEND = os.path.join(CACHE_PATH, "eval-queue-bk")
63
  EVAL_REQUESTS_PATH_BACKEND_SYNC = os.path.join(CACHE_PATH, "eval-queue-bk-sync")
64
  EVAL_RESULTS_PATH_BACKEND = os.path.join(CACHE_PATH, "eval-results-bk")
65
 
66
- DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 
63
  EVAL_REQUESTS_PATH_BACKEND_SYNC = os.path.join(CACHE_PATH, "eval-queue-bk-sync")
64
  EVAL_RESULTS_PATH_BACKEND = os.path.join(CACHE_PATH, "eval-results-bk")
65
 
66
+ DEVICE = "cuda:0" if torch.cuda.is_available() else "cpu"