space

EleutherAI · Aug 20, 2024 · 0aaaa32 · 0aaaa32
1 parent adc4aec
commit 0aaaa32
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/megatron/model/mamba/mamba.py b/megatron/model/mamba/mamba.py
@@ -3,6 +3,7 @@
 import torch
 import torch.nn as nn
 import torch.nn.functional as F
+
 try:
     from mamba_ssm.ops.selective_scan_interface import (
         selective_scan_ref,