Update losses.py

Author: anas-awadalla

open_flamingo/train/losses.py

@@ -57,7 +57,6 @@ def __call__(
         attention_mask: torch.Tensor,
         autocast: callable,
     ):
-        print(input_ids[0])
         # set up labels; language model is expected to handle shifting
         labels = input_ids.clone()
         labels[labels == tokenizer.pad_token_id] = -100
@@ -88,4 +87,4 @@ def unwrap_model(model):
     ):
         return model.module
     else:
-        return model
+        return model