mirror of https://github.com/rasbt/LLMs-from-scratch.git synced 2026-04-11 14:21:41 +08:00

History

Sebastian Raschka 3233ddc475 get rid of redundant memory profiler import (#744 )		2025-07-16 07:36:51 -05:00
..
memory-efficient-state-dict.ipynb	get rid of redundant memory profiler import (#744 )	2025-07-16 07:36:51 -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