Commit
·
b6bdcd4
1
Parent(s):
5e31360
Fix missing closing parenthesis
Browse filesThe code snippet did not work because of a missing parenthesis.
I'd recommend to mention the need to `pip install transformers einops torch`
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")
|