mirror of https://github.com/rasbt/LLMs-from-scratch.git synced 2026-04-13 15:01:40 +08:00

History

Sebastian Raschka c21bfe4a23 Add PyPI package (#576 ) * Add PyPI package * fixes * fixes		2025-03-23 19:28:49 -05:00
..
memory-efficient-state-dict.ipynb	Add PyPI package (#576 )	2025-03-23 19:28:49 -05:00
previous_chapters.py	Memory efficient weight loading (#401 )	2024-10-14 10:30:25 -05:00
README.md	Memory efficient weight loading (#401 )	2024-10-14 10:30:25 -05:00

Memory-efficient Model Weight Loading

This folder contains code to illustrate how to load model weights more efficiently

memory-efficient-state-dict.ipynb: contains code to load model weights via PyTorch's load_state_dict method more efficiently