Skip to content

Loading Model weights with fastsafetensors

Using fastsafetensors library enables loading model weights to GPU memory by leveraging GPU direct storage. See their GitHub repository for more details. For enabling this feature, set the environment variable USE_FASTSAFETENSOR to true