@@ -12,10 +12,10 @@ conda create -n llm python=3.11
12
12
conda activate llm
13
13
# below command will install intel_extension_for_pytorch==2.1.10+xpu as default
14
14
pip install --pre --upgrade ipex-llm[xpu] --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/us/
15
- pip install transformers==4.36.0 datasets
15
+ pip install transformers==4.45.0 " trl<0.12.0 " datasets
16
16
pip install fire peft==0.10.0
17
+ pip install bitsandbytes==0.45.1 scipy
17
18
pip install oneccl_bind_pt==2.1.100 --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/us/ # necessary to run distributed finetuning
18
- pip install bitsandbytes scipy
19
19
```
20
20
21
21
### 2. Configures OneAPI environment variables
@@ -57,14 +57,14 @@ python ./alpaca_qalora_finetuning.py \
57
57
58
58
### 5. Sample Output
59
59
``` log
60
- {'loss': 1.9231, 'learning_rate': 2.9999945367033285e-05, 'epoch': 0.0}
61
- {'loss': 1.8622, 'learning_rate': 2.9999781468531096e-05, 'epoch': 0.01}
62
- {'loss': 1.9043, 'learning_rate': 2.9999508305687345e-05, 'epoch': 0.01}
63
- {'loss': 1.8967, 'learning_rate': 2.999912588049185e-05, 'epoch': 0.01}
64
- {'loss': 1.9658, 'learning_rate': 2.9998634195730358e-05, 'epoch': 0.01}
65
- {'loss': 1.8386, 'learning_rate': 2.9998033254984483e-05, 'epoch': 0.02}
66
- {'loss': 1.809, 'learning_rate': 2.999732306263172e-05, 'epoch': 0.02}
67
- {'loss': 1.8552, 'learning_rate': 2.9996503623845395e-05, 'epoch': 0.02}
60
+ {'loss': 1.9231, 'learning_rate': 2.9999945367033285e-05, 'epoch': 0.0}
61
+ {'loss': 1.8622, 'learning_rate': 2.9999781468531096e-05, 'epoch': 0.01}
62
+ {'loss': 1.9043, 'learning_rate': 2.9999508305687345e-05, 'epoch': 0.01}
63
+ {'loss': 1.8967, 'learning_rate': 2.999912588049185e-05, 'epoch': 0.01}
64
+ {'loss': 1.9658, 'learning_rate': 2.9998634195730358e-05, 'epoch': 0.01}
65
+ {'loss': 1.8386, 'learning_rate': 2.9998033254984483e-05, 'epoch': 0.02}
66
+ {'loss': 1.809, 'learning_rate': 2.999732306263172e-05, 'epoch': 0.02}
67
+ {'loss': 1.8552, 'learning_rate': 2.9996503623845395e-05, 'epoch': 0.02}
68
68
1%|█ | 8/1164 [xx:xx<xx:xx:xx, xx s/it]
69
69
```
70
70
0 commit comments