search

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

improve FrankenSolver #391

Closed catwell closed 1 month ago

catwell commented 1 month ago

It now takes a Scheduler factory instead of a Scheduler. This lets the user potentially recreate the Scheduler on rebuild.

It also properly sets the device and dtype on rebuild, and it has better typing.