Babelscape
/

mrebel-large

text2text-generation

relation-extraction

Inference Endpoints

Model card Files Files and versions Community

PereLluis13 commited on Jun 20, 2023

Commit

5df3a7a

•

1 Parent(s): 628ac2e

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -166,7 +166,7 @@ model_inputs = tokenizer(text, max_length=256, padding=True, truncation=True, re
 generated_tokens = model.generate(
     model_inputs["input_ids"].to(model.device),
     attention_mask=model_inputs["attention_mask"].to(model.device),
-    decoder_start_token_id = self.tokenizer.convert_tokens_to_ids("tp_XX"),
     **gen_kwargs,
 )

 generated_tokens = model.generate(
     model_inputs["input_ids"].to(model.device),
     attention_mask=model_inputs["attention_mask"].to(model.device),
+    decoder_start_token_id = tokenizer.convert_tokens_to_ids("tp_XX"),
     **gen_kwargs,
 )