From 9890ae6403ecc0b9a6c9c4ca34ba604d5101f4ef Mon Sep 17 00:00:00 2001 From: Juarez Bochi Date: Thu, 21 Dec 2023 10:08:03 -0500 Subject: [PATCH] Remove unused import --- t5/t5.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/t5/t5.py b/t5/t5.py index 17f3f11a..45855136 100644 --- a/t5/t5.py +++ b/t5/t5.py @@ -5,7 +5,7 @@ from typing import List, Optional, Tuple import mlx.core as mx import mlx.nn as nn import numpy as np -from mlx.utils import tree_map, tree_flatten, tree_unflatten +from mlx.utils import tree_map, tree_unflatten from transformers import T5Config, AutoTokenizer