Update README.md
Browse files
README.md
CHANGED
|
@@ -32,6 +32,10 @@ This model was fine-tuned with reinforcement learning to extract structured info
|
|
| 32 |
You can use this model to extract structured information from text with the following code:
|
| 33 |
|
| 34 |
```python
|
|
|
|
|
|
|
|
|
|
|
|
|
| 35 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 36 |
|
| 37 |
# Load the model and tokenizer
|
|
|
|
| 32 |
You can use this model to extract structured information from text with the following code:
|
| 33 |
|
| 34 |
```python
|
| 35 |
+
!pip install -q transformers>=4.36.0 accelerate>=0.24.0 huggingface_hub>=0.18.0
|
| 36 |
+
!pip install -q bitsandbytes>=0.41.0
|
| 37 |
+
|
| 38 |
+
# No need to install notebook-login as it's included in huggingface_hub
|
| 39 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 40 |
|
| 41 |
# Load the model and tokenizer
|