somosnlp
/

NoticIA-7B

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

Iker commited on Mar 27

Commit

0f41a56

•

1 Parent(s): 80642cc

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -211,7 +211,7 @@ model = AutoModelForCausalLM.from_pretrained(
 )
 formatted_prompt = tokenizer.apply_chat_template(
-    [{"role": "user", "content": dataset[0]["prompt"]}],
     tokenize=False,
     add_generation_prompt=True,
 )

 )
 formatted_prompt = tokenizer.apply_chat_template(
+    [{"role": "user", "content": dataset[0]["pregunta"]}],
     tokenize=False,
     add_generation_prompt=True,
 )