Skip to content

Commit 9d6b60d

Browse files
committed
justpythonthings
Signed-off-by: Sertac Ozercan <sozercan@gmail.com>
1 parent 6bddfb6 commit 9d6b60d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pkg/aikit2llb/finetune/convert.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@ func Aikit2LLB(c *config.FineTuneConfig) llb.State {
4848
if c.Target == utils.TargetUnsloth {
4949
// installing unsloth and its dependencies
5050
// uv does not support installing xformers via unsloth pyproject
51-
state = state.Run(utils.Shf("pip install --upgrade pip uv && uv venv --system-site-packages && %[1]s && uv pip install packaging torch==2.1.0 ipython ninja packaging bitsandbytes setuptools wheel psutil && uv pip install flash-attn --no-build-isolation && python -m pip install 'unsloth[cu121_ampere] @ git+https://github.com/unslothai/unsloth.git@%[2]s'", sourceVenv, unslothCommitSHA)).Root()
51+
state = state.Run(utils.Shf("pip install --upgrade pip uv && uv venv --system-site-packages && %[1]s && uv pip install packaging torch==2.1.0 ipython ninja packaging bitsandbytes setuptools==69.5.1 wheel psutil && uv pip install flash-attn --no-build-isolation && python -m pip install 'unsloth[cu121_ampere] @ git+https://github.com/unslothai/unsloth.git@%[2]s'", sourceVenv, unslothCommitSHA)).Root()
5252

5353
version := version.Version
5454
if version == "" {

0 commit comments

Comments
 (0)