Improved ColoAttention interface to support flash attention 2. Solved #4322
Attach a file by drag & drop or click to upload
