github/gpt4all
1
0
Fork 0
mirror of https://github.com/nomic-ai/gpt4all.git synced 2025-07-07 12:31:48 +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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
gpt4all/utils/distributed_utils.py
View File

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