IlayMalinyak commited on
Commit
4a743a9
·
1 Parent(s): 9c22c74

import fix

Browse files
Files changed (1) hide show
  1. tasks/utils/models.py +0 -1
tasks/utils/models.py CHANGED
@@ -3,7 +3,6 @@ import torch.nn as nn
3
  from .Modules.conformer import ConformerEncoder, ConformerDecoder
4
  from .Modules.mhsa_pro import RotaryEmbedding, ContinuousRotaryEmbedding
5
  from .kan.fasterkan import FasterKAN
6
- from kan import KAN
7
 
8
  class ConvBlock(nn.Module):
9
  def __init__(self, args, num_layer) -> None:
 
3
  from .Modules.conformer import ConformerEncoder, ConformerDecoder
4
  from .Modules.mhsa_pro import RotaryEmbedding, ContinuousRotaryEmbedding
5
  from .kan.fasterkan import FasterKAN
 
6
 
7
  class ConvBlock(nn.Module):
8
  def __init__(self, args, num_layer) -> None: