mrm8488 dvilasuero commited on
Commit
d9486db
1 Parent(s): 5e31360

Fix missing import and closing parenthesis (#1)

Browse files

- Fix missing closing parenthesis (b6bdcd4dda0efd7160b84a74c3775bbc16e94be3)
- Fix add missing import (f7310b31a09ad92036a7dff9cc9fbfe9d56ddf75)


Co-authored-by: Daniel Vila <dvilasuero@users.noreply.huggingface.co>

Files changed (1) hide show
  1. README.md +2 -2
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
 
@@ -208,7 +208,7 @@ def generate(
208
  ):
209
 
210
  prompt = create_instruction(instruction, input, context)
211
- print(prompt.replace("### Respuesta:\n", "")
212
  inputs = tokenizer(prompt, return_tensors="pt")
213
  input_ids = inputs["input_ids"].to("cuda")
214
  attention_mask = inputs["attention_mask"].to("cuda")
 
169
 
170
  ```py
171
  import torch
172
+ from transformers import AutoModelForCausalLM, AutoTokenizer, AutoTokenizer, GenerationConfig
173
 
174
  model_id = "clibrain/lince-zero"
175
 
 
208
  ):
209
 
210
  prompt = create_instruction(instruction, input, context)
211
+ print(prompt.replace("### Respuesta:\n", ""))
212
  inputs = tokenizer(prompt, return_tensors="pt")
213
  input_ids = inputs["input_ids"].to("cuda")
214
  attention_mask = inputs["attention_mask"].to("cuda")