Update umt5_summary.py

umt5_summary.py

@@ -39,7 +39,7 @@ input_ids = tokenizer(
     max_length=512
 ).to(device)
 
-generated_tokens = model.module.generate(
+generated_tokens = model.generate(
     input_ids["input_ids"],
     attention_mask=input_ids["attention_mask"],
     max_length=32,