Delete chatNT_config.py

chatNT_config.py

@@ -1,50 +0,0 @@
-from dataclasses import dataclass
-
-from transformers import PretrainedConfig
-
-from genomics_research.biobrain_p1.porting_to_pytorch.configs.esm_config import (
-    ESMTransformerConfig,
-)
-from genomics_research.biobrain_p1.porting_to_pytorch.configs.gpt_config import (
-    GptConfig,
-)
-from genomics_research.biobrain_p1.porting_to_pytorch.configs.perceiver_resampler_config import (  # noqa
-    PerceiverResamplerConfig,
-)
-
-
-@dataclass
-class ChatNTConfig(PretrainedConfig):
-    model_type = "ChatNT"
-
-    def __init__(self, **kwargs):  # type: ignore
-        self.gpt_config: GptConfig = kwargs.get("gpt_config", GptConfig(32000, 3))
-        self.esm_config: ESMTransformerConfig = kwargs.get(
-            "esm_config", ESMTransformerConfig(4000, 1, 4)
-        )
-        self.perceiver_resampler_config: PerceiverResamplerConfig = kwargs.get(
-            "perceiver_resampler_config", PerceiverResamplerConfig()
-        )
-        self.seq_token_id: int = kwargs.get("seq_token_id", 32000)
-        self.bio_pad_token_id: int = kwargs.get("bio_pad_token_id", 1)
-        self.english_pad_token_id: int = kwargs.get("english_pad_token_id", 2)
-        super().__init__(**kwargs)
-
-    def to_dict(self):  # type: ignore
-        print("(debug) Going into ChatNTConfig to_dict")
-        output = super().to_dict()
-
-        def serialize(obj):  # type: ignore
-            return obj.to_dict() if hasattr(obj, "to_dict") else vars(obj)
-
-        print("(debug) Before serialize gpt_config to_dict")
-        output["gpt_config"] = serialize(self.gpt_config)  # type: ignore
-        print("(debug) Before serialize esm_config to_dict")
-        output["esm_config"] = serialize(self.esm_config)  # type: ignore
-        print("(debug) Before serialize perceiver_resampler_config to_dict")
-        output["perceiver_resampler_config"] = serialize(  # type: ignore
-            self.perceiver_resampler_config
-        )
-        print("(debug) after serializing all ")
-        print("(debug) output : ", output)
-        return output