tacit/nanochat
1
0
Fork 0
mirror of https://github.com/karpathy/nanochat.git synced 2025-12-06 12:22:18 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Revert "Refactor: Improve DDP detection in common.py"

Browse Source 
This reverts commit b724190f2a.
This commit is contained in:
Azekowka 2025-10-14 17:04:41 +05:00
parent b724190f2a
commit 1a45e5b78a
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
nanochat/common.py
View File

@ -76,8 +76,7 @@ def print_banner():
print0(banner) print0(banner)
def is_ddp(): def is_ddp():
if dist.is_initialized(): # TODO is there a proper way
return True
return int(os.environ.get('RANK', -1)) != -1 return int(os.environ.get('RANK', -1)) != -1
def get_dist_info(): def get_dist_info():