Update chatNT.py
Browse files
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
|