Cleanup

toolbox/models/embedding/embedder/embedder_type.py

@@ -1,17 +1,13 @@
 from enum import Enum
 from toolbox.models.embedding.embedder.esm2_embedder import ESM2Embedder
 from toolbox.models.embedding.embedder.esmc_embedder import ESMCEmbedder
-from toolbox.models.embedding.embedder.esmc_performance_embedder import ESMCPerformanceEmbedder
-from toolbox.models.embedding.embedder.esm2_performance_embedder import ESM2PerformanceEmbedder
 from toolbox.models.embedding.embedder.base_embedder import BaseEmbedder
 
 class EmbedderType(Enum):
     ESM2_T30_150M = ("esm2_t30_150M_UR50D", ESM2Embedder, 640)
     ESM2_T33_650M = ("esm2_t33_650M_UR50D", ESM2Embedder, 1280)
     ESMC_300M = ("esmc_300m", ESMCEmbedder, 960)
     ESMC_600M = ("esmc_600m", ESMCEmbedder, 1152)
-    ESMC_600M_PERFORMANCE = ("esmc_600m_performance", ESMCPerformanceEmbedder, 1152)
-    ESM2_T33_650M_PERFORMANCE = ("esm2_t33_650M_UR50D_performance", ESM2PerformanceEmbedder, 1280)
 
     def __init__(self, value, embedder_class: BaseEmbedder, embedding_size: int):
         self._value_ = value