set TORCH_CUDA_ARCH_LIST correctly

upgrade liger dep to 0.6.3
2025-10-29 12:59:26 -04:00 · 2025-10-27 14:49:09 -04:00
2 changed files with 2 additions and 2 deletions
--- a/cicd/Dockerfile.jinja
+++ b/cicd/Dockerfile.jinja
@@ -1,6 +1,6 @@
 FROM axolotlai/axolotl-base:{{ BASE_TAG }}

-ENV TORCH_CUDA_ARCH_LIST="7.0 7.5 8.0 8.6+PTX"
+ENV TORCH_CUDA_ARCH_LIST="7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX"
 ENV AXOLOTL_EXTRAS="{{ AXOLOTL_EXTRAS }}"
 ENV AXOLOTL_ARGS="{{ AXOLOTL_ARGS }}"
 ENV CUDA="{{ CUDA }}"
--- a/requirements.txt
+++ b/requirements.txt
@@ -5,7 +5,7 @@ bitsandbytes==0.47.0
 triton>=3.0.0
 mamba-ssm==1.2.0.post1
 xformers>=0.0.23.post1
-liger-kernel==0.6.1
+liger-kernel==0.6.3
 # END section

 packaging==23.2
Author	SHA1	Message	Date
Wing Lian	9ee7ce5c85	set TORCH_CUDA_ARCH_LIST correctly	2025-10-29 12:59:26 -04:00
Wing Lian	a41ca4d06f	upgrade liger dep to 0.6.3	2025-10-27 14:49:09 -04:00