fastsafetensor.md 414 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