Update chatNT.py

chatNT.py

@@ -588,7 +588,9 @@ class TorchMultiOmicsModel(PreTrainedModel):
     config_class = ChatNTConfig
 
     def __init__(self, config: ChatNTConfig) -> None:
+        print("(debug) Entering in class")
         if isinstance(config, dict):
+            print("(debug) going in if condition")
             # If config is a dictionary instead of ChatNTConfig (which can happen
             # depending how the config was saved), we convert it to the config
             config["gpt_config"]["rope_config"] = RotaryEmbeddingConfig(
@@ -596,10 +598,15 @@ class TorchMultiOmicsModel(PreTrainedModel):
             )
             config["gpt_config"] = GptConfig(**config["gpt_config"])
             config["esm_config"] = ESMTransformerConfig(**config["esm_config"])
+            print("(debug) Type esm_config : ", type(config["esm_config"]))
+            print("(debug) esm_config : ", config["esm_config"])
             config["perceiver_resampler_config"] = PerceiverResamplerConfig(
                 **config["perceiver_resampler_config"]
             )
             config = ChatNTConfig(**config)  # type: ignore
+            print("(debug) Type config : ", type(config))
+
+        print("(debug) config : ", config)
 
         super().__init__(config=config)
         self.gpt_config = config.gpt_config