dvilasuero HF staff commited on
Commit
f7310b3
1 Parent(s): b6bdcd4

Fix add missing import

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -169,7 +169,7 @@ Use the code below to get started with LINCE-ZERO!
169
 
170
  ```py
171
  import torch
172
- from transformers import AutoModelForCausalLM, AutoTokenizer, AutoTokenizer
173
 
174
  model_id = "clibrain/lince-zero"
175
 
 
169
 
170
  ```py
171
  import torch
172
+ from transformers import AutoModelForCausalLM, AutoTokenizer, AutoTokenizer, GenerationConfig
173
 
174
  model_id = "clibrain/lince-zero"
175