# Additional requirements for GPU environments
# Use this file if you have a CUDA-enabled GPU and want to use flash-attn
flash_attn