Skip to content

GitLab

Commit 02c43ac5 authored by wanglch's avatar wanglch
Browse files

Update umt5_summary.py

parent 0322f391
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
umt5_summary.py
View file @ 02c43ac5
...@@ -39,7 +39,7 @@ input_ids = tokenizer( ...@@ -39,7 +39,7 @@ input_ids = tokenizer(
max_length=512 max_length=512
).to(device) ).to(device)
generated_tokens = model.module.generate( generated_tokens = model.generate(
input_ids["input_ids"], input_ids["input_ids"],
attention_mask=input_ids["attention_mask"], attention_mask=input_ids["attention_mask"],
max_length=32, max_length=32,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment