Update README.md
Browse files
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"
|