fastsafetensor.md 414 Bytes
Newer Older
1
Loading model weights with fastsafetensors
2
3
===================================================================

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