Merge pull request FlagAI-Open#360 from zhiyongLiu1114/master

[FIX_BUG] when torch.__version__ >=2, sys should be imported
jkhomme4 · Jun 12, 2023 · fa9b5a3 · fa9b5a3
2 parents 128a809 + 2746a76
commit fa9b5a3
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/flagai/mpu/grads.py b/flagai/mpu/grads.py
@@ -20,6 +20,7 @@
 # repo: https://github.com/pytorch/pytorch
 
 import torch
+import sys
 if torch.__version__ >= "2" and sys.platform != "win32":
     from torch import inf
 else: