mrs83 commited on
Commit
f9e3cdb
·
verified ·
1 Parent(s): d7873b0

Update dsrn_memory_block.py

Browse files
Files changed (1) hide show
  1. dsrn_memory_block.py +1 -1
dsrn_memory_block.py CHANGED
@@ -46,7 +46,7 @@ import torch.nn as nn
46
  import torch.nn.functional as F
47
 
48
  # Canonical imports from the parent Echo-DSRN package.
49
- from echo_dsrn.modeling_echo import HymbaRMSNorm, dsrn_parallel_scan, rms_norm_fn
50
 
51
  from .configuration_hybrid import HybridEchoConfig
52
 
 
46
  import torch.nn.functional as F
47
 
48
  # Canonical imports from the parent Echo-DSRN package.
49
+ from .modeling_echo import HymbaRMSNorm, dsrn_parallel_scan, rms_norm_fn
50
 
51
  from .configuration_hybrid import HybridEchoConfig
52