diff --git a/nvidia/unsloth/README.md b/nvidia/unsloth/README.md index 19cf0c7..f62e1aa 100644 --- a/nvidia/unsloth/README.md +++ b/nvidia/unsloth/README.md @@ -85,7 +85,7 @@ docker run --gpus all --ulimit memlock=-1 -it --ulimit stack=67108864 --entrypoi ## Step 4. Install dependencies inside Docker ```bash -pip install transformers peft datasets "trl==0.19.1" +pip install transformers peft "datasets==4.3.0" "trl==0.19.1" pip install --no-deps unsloth unsloth_zoo pip install hf_transfer ```