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
|