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")
|