Upload modeling_fastesm.py with huggingface_hub

modeling_fastesm.py

@@ -25,7 +25,12 @@ from transformers.models.esm.modeling_esm import (
     EsmClassificationHead,
 )
 
-from .embedding_mixin import EmbeddingMixin
+try:
+    # when used from AutoModel, these are in the same directory
+    from .embedding_mixin import EmbeddingMixin, Pooler
+except:
+    # when running from our repo, these are in the base directory
+    from embedding_mixin import EmbeddingMixin, Pooler
 
 
 def _create_pad_block_mask(attention_mask_2d: torch.Tensor):