Yanisadel commited on
Commit
8d737ee
·
1 Parent(s): 6b068b7

Update chatNT.py

Browse files
Files changed (1) hide show
  1. chatNT.py +0 -8
chatNT.py CHANGED
@@ -588,7 +588,6 @@ class TorchMultiOmicsModel(PreTrainedModel):
588
  config_class = ChatNTConfig
589
 
590
  def __init__(self, config: ChatNTConfig) -> None:
591
- print("(debug) Entering in class")
592
  if isinstance(config, dict):
593
  # If config is a dictionary instead of ChatNTConfig (which can happen
594
  # depending how the config was saved), we convert it to the config
@@ -614,13 +613,6 @@ class TorchMultiOmicsModel(PreTrainedModel):
614
 
615
  if isinstance(config.perceiver_resampler_config, dict):
616
  config.perceiver_resampler_config = PerceiverResamplerConfig(**config.perceiver_resampler_config)
617
-
618
- print("(debug) config : ", config)
619
- print("(debug) config type : ", type(config))
620
- print("(debug) gpt config : ", config.gpt_config)
621
- print("(debug) gpt config type : ", type(config.gpt_config))
622
- print("(debug) esm config : ", config.esm_config)
623
- print("(debug) esm config type : ", type(config.esm_config))
624
 
625
  super().__init__(config=config)
626
  self.gpt_config = config.gpt_config
 
588
  config_class = ChatNTConfig
589
 
590
  def __init__(self, config: ChatNTConfig) -> None:
 
591
  if isinstance(config, dict):
592
  # If config is a dictionary instead of ChatNTConfig (which can happen
593
  # depending how the config was saved), we convert it to the config
 
613
 
614
  if isinstance(config.perceiver_resampler_config, dict):
615
  config.perceiver_resampler_config = PerceiverResamplerConfig(**config.perceiver_resampler_config)
 
 
 
 
 
 
 
616
 
617
  super().__init__(config=config)
618
  self.gpt_config = config.gpt_config