Upload esm2-flash-3B (ESM2 with flash attention)

model-00003-of-00003.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ae6ca76ffa01e0c1a13d0b974c64e1f2a76bfda734ea40149c46a9fdc42b4de9
+oid sha256:b5de1ca0538d20d96d513315bdeda787ec9ce8a1825add5a697820d3c337ca4a
 size 1494870972

modeling_esm2_flash.py

@@ -27,10 +27,7 @@ from transformers.modeling_outputs import (
 )
 from transformers.modeling_utils import PreTrainedModel
 
-try:
-    from .configuration_esm2_flash import Esm2FlashConfig
-except ImportError:
-    from configuration_esm2_flash import Esm2FlashConfig
+from .configuration_esm2_flash import Esm2FlashConfig
 
 try:
     from flash_attn.flash_attn_interface import flash_attn_varlen_func