|
78e0c7eec9
|
Specify cu128 Pytorch installation (#530)
Co-authored-by: Edenzzzz <wtan45@wisc.edu>
Co-authored-by: Wenxuan Tan <wenxuan.tan@wisc.edu>
|
2025-06-18 20:02:50 -05:00 |
|
|
Yongqi ChenandWill Lin
|
55074bda76
|
[CI] Add STA-inference/VSA-training test (#527)
Co-authored-by: Will Lin <wlsaidhi@gmail.com>
|
2025-06-17 21:13:06 -07:00 |
|
|
William Lin
|
de65bec2b7
|
[Ci] add sta and vsa install to docker image (#528)
|
2025-06-17 18:09:48 -07:00 |
|
|
Kevin Lin
|
d71a4ebffc
|
[CI] Update Docker image to flash-attn 2.8.0 / CUDA 12.8 (#524)
|
2025-06-16 17:48:23 -07:00 |
|
|
Kevin Lin
|
048d55ccbb
|
[CI] Add new images for different Python versions (#377)
|
2025-05-05 10:28:33 -07:00 |
|