fastsafetensor.md 418 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
For enabling this feature, set the environment variable ``USE_FASTSAFETENSOR`` to ``true``