From 689eda9937b332b78d325c3fc4fe59119883edd1 Mon Sep 17 00:00:00 2001
From: Juarez Bochi <juarez.bochi@grammarly.com>
Date: Mon, 18 Dec 2023 08:00:01 -0500
Subject: [PATCH] Fix T5.__call__

---
 t5/t5.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/t5/t5.py b/t5/t5.py
index dcdcfbfe..c137e0c1 100644
--- a/t5/t5.py
+++ b/t5/t5.py
@@ -314,7 +314,7 @@ class T5(nn.Module):
         inputs: mx.array,
         decoder_inputs: mx.array,
     ):
-        return decode(decoder_inputs, encode(inputs))[0]
+        return self.decode(decoder_inputs, self.encode(inputs))[0]
 
 
 def generate(