remove debug logs

Author: 92HyungjunOh

src/transformers/models/gpt2/modeling_gpt2_moreh.py

@@ -1302,13 +1302,6 @@ class GPT2LMHeadModelMoreh(GPT2PreTrainedModel):
 
     def __init__(self, config):
         super().__init__(config)
-        print("GPT2LMHeadModelMoreh ##################################")
-        if config.moreh_config is not None:
-            print("config.moreh_config")
-            for key, value in config.moreh_config.items():
-                print(f"\t {key}, {value}")
-        else:
-            print("config.moreh_config is None")
         self.transformer = GPT2Model(config)
         self.lm_head = nn.Linear(config.n_embd, config.vocab_size, bias=False)
 

src/transformers/models/mistral/modeling_mistral_moreh.py

@@ -1024,7 +1024,6 @@ def forward(
 
             hidden_states = layer_outputs[0]
             if layer_idx in self.moreh_pipeline_layers:
-                print(f"Set pipe in mistral L : {layer_idx}")
                 hidden_states = torch.moreh.pipeline_assign(hidden_states)
 
             if use_cache:
@@ -1165,14 +1164,6 @@ class MistralForCausalLMMoreh(MistralPreTrainedModel):
 
     def __init__(self, config):
         super().__init__(config)
-        print("MistralForCausalLMMoreh #########################################")
-        if config.moreh_config is not None:
-            print("config.moreh_config")
-            for key, value in config.moreh_config.items():
-                print(f"\t {key}, {value}")
-        else:
-            print("config.moreh_config is None")
-
         self.model = MistralModel(config)
         self.vocab_size = config.vocab_size
         self.lm_head = nn.Linear(config.hidden_size, config.vocab_size, bias=False)