From 710686f446f02286c858c11f052acb87c306ddd2 Mon Sep 17 00:00:00 2001 From: Gustavo de Rosa Date: Wed, 17 Apr 2024 13:52:46 +0000 Subject: [PATCH] Update README.md --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 3888c1a..aef4817 100644 --- a/README.md +++ b/README.md @@ -99,7 +99,7 @@ inputs = tokenizer('''def print_prime(n): """''', return_tensors="pt", return_attention_mask=False) outputs = model.generate(**inputs, max_length=200) -text = tokenizer.batch_decode(outputs[0]) +text = tokenizer.batch_decode(outputs)[0] print(text) ```