Soh19 commited on
Commit
0397d4e
·
verified ·
1 Parent(s): fd1fbcd

Update README.md

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