fix: str should be used not int when setting env variables (huggi…

…ngface#31272)
mesolitica · Jun 6, 2024 · 9e9679c · 9e9679c
1 parent 9ef93fc
commit 9e9679c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/transformers/trainer_utils.py b/src/transformers/trainer_utils.py
@@ -72,8 +72,8 @@ def enable_full_determinism(seed: int, warn_only: bool = False):
         os.environ["CUDA_LAUNCH_BLOCKING"] = "1"
         os.environ["CUBLAS_WORKSPACE_CONFIG"] = ":16:8"
         # The environment variable required to enable deterministic mode on Ascend NPUs.
-        os.environ["ASCEND_LAUNCH_BLOCKING"] = 1
-        os.environ["HCCL_DETERMINISTIC"] = 1
+        os.environ["ASCEND_LAUNCH_BLOCKING"] = "1"
+        os.environ["HCCL_DETERMINISTIC"] = "1"
         torch.use_deterministic_algorithms(True, warn_only=warn_only)
 
         # Enable CUDNN deterministic mode