tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

release version 0.5.1 (#2082)

Browse Source
This commit is contained in:
Wing Lian
2024-11-19 10:35:59 -05:00
committed by GitHub
parent ec59d4cb83
commit 6679e20f47
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
setup.py
View File

@@ -96,7 +96,7 @@ install_requires, dependency_links = parse_requirements()
setup( setup(
name="axolotl", name="axolotl",
version="0.5.0", version="0.5.1",
description="LLM Trainer", description="LLM Trainer",
long_description="Axolotl is a tool designed to streamline the fine-tuning of various AI models, offering support for multiple configurations and architectures.", long_description="Axolotl is a tool designed to streamline the fine-tuning of various AI models, offering support for multiple configurations and architectures.",
package_dir={"": "src"}, package_dir={"": "src"},
@@ -108,7 +108,7 @@ setup(
"flash-attn==2.7.0.post2", "flash-attn==2.7.0.post2",
], ],
"deepspeed": [ "deepspeed": [
"deepspeed==0.14.4", "deepspeed==0.15.4",
"deepspeed-kernels", "deepspeed-kernels",
], ],
"mamba-ssm": [ "mamba-ssm": [