Fix typing issues coming from torch 2.4 version

finegrain-ai / refiners

A microframework on top of PyTorch with first-class citizen APIs for foundation model adaptation

https://refine.rs

MIT License

584 stars 43 forks source link

Closed limiteinductive closed 1 month ago

limiteinductive commented 1 month ago

With the release of PyTorch 2.4, some typing changed, most notably:

different imports for the torch.optim package
they fixed the nn.Parameter typing issue introduced in 2.2 (see https://github.com/pytorch/pytorch/issues/118736)