Update README.md
Browse files
README.md
CHANGED
@@ -12,8 +12,18 @@ library_name: peft
|
|
12 |
- Training method: Instruction Fine-tuning + LoRA + fp16
|
13 |
- Task: Sentiment Analysis
|
14 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
15 |
|
16 |
-
## Try the model
|
17 |
``` python
|
18 |
from transformers import AutoModel, AutoTokenizer, AutoModelForCausalLM, LlamaForCausalLM, LlamaTokenizerFast
|
19 |
from peft import PeftModel # 0.5.0
|
|
|
12 |
- Training method: Instruction Fine-tuning + LoRA + fp16
|
13 |
- Task: Sentiment Analysis
|
14 |
|
15 |
+
## Packages
|
16 |
+
``` python
|
17 |
+
!pip install transformers==4.32.0 peft==0.5.0
|
18 |
+
!pip install sentencepiece
|
19 |
+
!pip install accelerate
|
20 |
+
!pip install torch
|
21 |
+
!pip install peft
|
22 |
+
!pip install datasets
|
23 |
+
!pip install bitsandbytes
|
24 |
+
```
|
25 |
|
26 |
+
## Try the model in Google Colab
|
27 |
``` python
|
28 |
from transformers import AutoModel, AutoTokenizer, AutoModelForCausalLM, LlamaForCausalLM, LlamaTokenizerFast
|
29 |
from peft import PeftModel # 0.5.0
|