From 49df0517f2a24296bbd55156931bf09ce403c22e Mon Sep 17 00:00:00 2001 From: Nicolas Frugoni Date: Sat, 16 Dec 2023 02:59:10 +0000 Subject: [PATCH] remove broadcat import --- 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