fastsafetensor.md 416 Bytes
Newer Older
1
2
3
Loading Model weights with fastsafetensors
===================================================================

Reid's avatar
Reid committed
4
Using fastsafetensors library enables loading model weights to GPU memory by leveraging GPU direct storage. See [their GitHub repository](https://github.com/foundation-model-stack/fastsafetensors) for more details.
5
6

To enable this feature, use the ``--load-format fastsafetensors`` command-line argument