From a5726e2b05d78b14b86d28d82a41fbedb20aecf1 Mon Sep 17 00:00:00 2001 From: Martin Shkreli Date: Fri, 15 Dec 2023 13:27:14 -0500 Subject: [PATCH] Update transformer.py remove broken reference, which is not used --- tortoise/models/transformer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tortoise/models/transformer.py b/tortoise/models/transformer.py index 707e9eb..d16cbc9 100644 --- a/tortoise/models/transformer.py +++ b/tortoise/models/transformer.py @@ -3,7 +3,7 @@ from functools import partial import torch import torch.nn.functional as F from einops import rearrange -from rotary_embedding_torch import RotaryEmbedding, broadcat +from rotary_embedding_torch import RotaryEmbedding from torch import nn