mirror of
https://github.com/NVIDIA/dgx-spark-playbooks.git
synced 2026-04-24 10:53:52 +00:00
update instructions for unsloth training
This commit is contained in:
parent
e4ca6905af
commit
a363f5964e
@ -89,9 +89,10 @@ pip install transformers peft datasets "trl==0.19.1"
|
|||||||
pip install --no-deps unsloth unsloth_zoo
|
pip install --no-deps unsloth unsloth_zoo
|
||||||
```
|
```
|
||||||
|
|
||||||
## Step 5. Build and install bitsandbytes inside Docker
|
## Step 5. Build and install bitsandbytes and hf_transfer inside Docker
|
||||||
```bash
|
```bash
|
||||||
pip install --no-deps bitsandbytes
|
pip install --no-deps bitsandbytes
|
||||||
|
pip install hf_transfer
|
||||||
```
|
```
|
||||||
|
|
||||||
## Step 6. Create Python test script
|
## Step 6. Create Python test script
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user