From 181a9398d102413131f666da7a148c0992e7fa3f Mon Sep 17 00:00:00 2001
From: xiaonengmiao <hlibb@connect.ust.hk>
Date: Tue, 5 Nov 2024 17:34:20 +0800
Subject: [PATCH] update distrib.py

---
 encodec/distrib.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/encodec/distrib.py b/encodec/distrib.py
index b1662d8..695a1c2 100644
--- a/encodec/distrib.py
+++ b/encodec/distrib.py
@@ -87,7 +87,7 @@ def sync_buffer(buffers, average=True):
     for buffer, handle in handles:
         handle.wait()
         if average:
-            buffer.data /= world_size
+            buffer.data /= world_size()
 
 
 def sync_grad(params):