changed device number back to 1

tum-ai · Jan 16, 2025 · 75aac1f · 75aac1f
1 parent e4acdd9
commit 75aac1f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/ntl/run_language_modeling.py b/src/ntl/run_language_modeling.py
@@ -8,7 +8,7 @@
 import sys
 import os
 sys.path.append(".")
-os.environ["CUDA_VISIBLE_DEVICES"] = "2"
+os.environ["CUDA_VISIBLE_DEVICES"] = "1"
 
 import time
 import json