fix: import error

congchan · Nov 11, 2023 · f3523a1 · f3523a1
1 parent da45035
commit f3523a1
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/llm/data/utils.py b/llm/data/utils.py
@@ -1,6 +1,8 @@
 """data utils"""
 
 import json
+import logging
+from torch.distributed import get_rank, is_initialized
 
 
 def read_text(path, debug=None):