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

Fix missing closing parenthesis

Browse files

The code snippet did not work because of a missing parenthesis.

I'd recommend to mention the need to `pip install transformers einops torch`

Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -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")
 
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")