This commit is contained in:
geronimi73 2025-10-15 10:14:22 -07:00 committed by GitHub
commit 6a31d9450b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

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