github/gpt4all
1
0
Fork 0
mirror of https://github.com/nomic-ai/gpt4all.git synced 2025-09-02 09:06:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: allow for print for fns that are used in both dist and single

Browse Source
This commit is contained in:
Zach Nussbaum
2023-04-25 20:34:12 +00:00
parent 7832707c37
commit 586a8abc06
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
gpt4all/utils/distributed_utils.py
View File

@@ -2,5 +2,8 @@ import torch.distributed as dist
def rank0_print(msg):
if dist.get_rank() == 0:
if dist.is_initialized():
if dist.get_rank() == 0:
print(msg)
else:
print(msg)