sepal commited on
Commit
4e6eb9d
β€’
1 Parent(s): 19216b2

Add missing import to example in readme.

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -54,7 +54,7 @@ TBA
54
  ### Example of usage πŸ‘©β€πŸ’»
55
  ```py
56
  import torch
57
- from transformers import AutoModelForCausalLM, AutoTokenizer, AutoTokenizer
58
 
59
  model_id = "mrm8488/falcoder-7b"
60
 
 
54
  ### Example of usage πŸ‘©β€πŸ’»
55
  ```py
56
  import torch
57
+ from transformers import AutoModelForCausalLM, AutoTokenizer, AutoTokenizer, GenerationConfig
58
 
59
  model_id = "mrm8488/falcoder-7b"
60