Skip to content

fix: pin hivemind>=1.1.11 for PyTorch 2.3+ compatibility#44

Open
vominh1919 wants to merge 1 commit intoPrimeIntellect-ai:mainfrom
vominh1919:fix/hivemind-pytorch-compat
Open

fix: pin hivemind>=1.1.11 for PyTorch 2.3+ compatibility#44
vominh1919 wants to merge 1 commit intoPrimeIntellect-ai:mainfrom
vominh1919:fix/hivemind-pytorch-compat

Conversation

@vominh1919
Copy link
Copy Markdown

Fixes #41

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).

Changes:

  • Pin hivemind to >=1.1.11 instead of a specific git commit that is incompatible with PyTorch 2.3+

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 PrimeIntellect-ai#41
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

1 participant