RecodeX 重构消息,Flax模型序列化可使用nnx.to_flat_state将嵌套状态扁平化,再通过Safetensors的flax API存储为点分隔键的简单字典格式。
Flax模型序列化可使用nnx.to_flat_state将嵌套状态扁平化,再通...
原始信息来源gilesthomas.comgilesthomas.com
查看原文 ↗RecodeX 重构消息,Flax模型序列化可使用nnx.to_flat_state将嵌套状态扁平化,再通过Safetensors的flax API存储为点分隔键的简单字典格式。