ByFinTech commited on
Commit
7c183e8
1 Parent(s): 68f885e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +11 -1
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