frozenleaves/trl
1
0
Fork 0
mirror of https://github.com/huggingface/trl.git synced 2025-10-20 18:43:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

Bump liger-kernel to fix grad acc and more features (#2333)

Browse Source 
Co-authored-by: Kashif Rasul <kashif.rasul@gmail.com>
This commit is contained in:
Byron Hsu
2024-11-08 03:16:33 -08:00
committed by GitHub
parent ac77c09223
commit c86b51cd12
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
setup.py
View File

@ -86,7 +86,7 @@ EXTRAS = {
"deepspeed": ["deepspeed>=0.14.4; sys_platform != 'win32'"],
"diffusers": ["diffusers>=0.18.0"],
# liger-kernel depends on triton, which is only available on Linux https://github.com/triton-lang/triton#compatibility
"liger": ["liger-kernel>=0.2.1; sys_platform != 'win32'"],
"liger": ["liger-kernel>=0.4.0; sys_platform != 'win32'"],
"llm_judge": ["openai>=1.23.2", "llm-blender>=0.0.2"],
"peft": ["peft>=0.8.0"],
"quantization": ["bitsandbytes"],