Update transformer.py

remove broken reference, which is not used
This commit is contained in:
Martin Shkreli 2023-12-15 13:27:14 -05:00 committed by GitHub
parent 1e061bc675
commit a5726e2b05
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