PEFT
flan
opt
crumb commited on
Commit
f77aea8
1 Parent(s): e31c3e5

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -0
README.md CHANGED
@@ -22,6 +22,7 @@ The model reaches a train ppl of 4.36 and an eval ppl of 4.32.
22
  ### Inference Example (Chain-of-Thought prompt):
23
 
24
  ```python
 
25
  from peft import PeftModel, PeftConfig
26
  from transformers import AutoModelForCausalLM, AutoTokenizer
27
  peft_model_id = "crumb/FLAN-OPT-6.7b-LoRA"
 
22
  ### Inference Example (Chain-of-Thought prompt):
23
 
24
  ```python
25
+ # %pip install -qq transformers git+https://github.com/huggingface/peft accelerate bitsandbytes
26
  from peft import PeftModel, PeftConfig
27
  from transformers import AutoModelForCausalLM, AutoTokenizer
28
  peft_model_id = "crumb/FLAN-OPT-6.7b-LoRA"