remove broadcat import

This commit is contained in:
Nicolas Frugoni 2023-12-16 02:59:10 +00:00 committed by GitHub
parent 1e061bc675
commit 49df0517f2
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -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