mirror of https://github.com/tracel-ai/burn.git
688958ee74
* Enhance PyTorchRecorder to pass top level key to extract state_dict This is needed for Whisper weight pt files. * Fix missing hyphens * Move top-level-key test under crates * Add sub-crates as members of workspace * Update Cargo.lock * Add accidentally omitted line during merge |
||
|---|---|---|
| .. | ||
| export_weights.py | ||
| mod.rs | ||
| top_level_key.pt | ||