Upload EnCodon

modeling_encodon.py

@@ -39,14 +39,13 @@ import torch.nn as nn
 from einops import rearrange, einsum
 from transformers.pytorch_utils import Conv1D
 
-from math import pi
 
 import torch
 from torch.amp import autocast
-from torch import nn, einsum, broadcast_tensors, Tensor
+from torch import nn, einsum, Tensor
 
 from einops import rearrange, repeat
-from typing import Optional, Union, Literal
+from typing import Optional
 
 
 def rotate_half(x):