cerebras.modelzoo.tools.checkpoint_converters.mup#
Functions
Classes
Transforms a CS 2.2 muP config to a CS 2.3 muP config with the reworked params. |
|
Transforms a CS 2.3 and onwards muP config to a CS sP config. |
|
Transforms a CS 2.2 and before muP config to a CS sP config. |
|
Transforms a T5 CS muP checkpoint into a T5 CS sP checkpoint. |
|
Transforms a CS 2.3 and onwards muP checkpoint into a CS sP checkpoint. |
|
Transforms a CS 2.2 and older muP checkpoints into a CS sP checkpoint. |