Update model/ea_model.py

model/ea_model.py

@@ -160,8 +160,6 @@ class EaModel(nn.Module):
 
         with torch.inference_mode():
             # Pass input through the base model
-            print("Weight is on device:", self.base_model.base_model.embed_tokens.weight.device)
-            print("Input is on device:", input_ids.device)
             outputs = self.base_model.model(
                 input_ids=input_ids,
                 attention_mask=attention_mask,