From 526dfaf878600bfb977fe1568cdbdb62567e336f Mon Sep 17 00:00:00 2001 From: vominh1919 Date: Sat, 18 Apr 2026 00:15:10 +0700 Subject: [PATCH] fix: pin hivemind>=1.1.11 for PyTorch 2.3+ compatibility The old pinned hivemind commit imports _refresh_per_optimizer_state from torch.cuda.amp.grad_scaler, which was removed/renamed in PyTorch 2.3+. hivemind 1.1.11+ includes a fix that checks the torch version and imports from the correct location (torch.amp for PyTorch >= 2.3.0). Fixes #41 --- requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index 8cd6920..84325fd 100644 --- a/requirements.txt +++ b/requirements.txt @@ -4,6 +4,6 @@ wandb>=0.16.4 cyclopts>=2.6.1 fsspec[gcs]>=2024.3.1 torch==2.3.1 -hivemind @ git+https://github.com/learning-at-home/hivemind.git@213bff9 +hivemind>=1.1.11 pydantic_config @ git+https://github.com/samsja/pydantic_config.git@8e19e05