mirror of
https://github.com/rasbt/LLMs-from-scratch.git
synced 2025-08-13 03:01:59 +00:00
6 lines
291 B
Markdown
6 lines
291 B
Markdown
|
# Memory-efficient Model Weight Loading
|
||
|
|
|
||
|
|
This folder contains code to illustrate how to load model weights more efficiently
|
||
|
|
|
||
|
|
- [memory-efficient-state-dict.ipynb](memory-efficient-state-dict.ipynb): contains code to load model weights via PyTorch's `load_state_dict` method more efficiently
|