Update modeling_qwen2_nomic_vision.py

modeling_qwen2_nomic_vision.py

@@ -1207,9 +1207,9 @@ class Qwen2NomicVisionForCausalLM(Qwen2NomicVisionPreTrainedModel, GenerationMix
 
         loss = None
         if labels is not None:
-            label_image = self.lm_head(outputs[-1][:, -num_logits_to_keep:, :])
-            print(label_image.shape, labels.shape)
-            labels = torch.concat([label_image, labels], dim=1)
+            # label_image = self.lm_head(outputs[-1][:, -num_logits_to_keep:, :])
+            # print(label_image.shape, labels.shape)
+            # labels = torch.concat([label_image, labels], dim=1)
             loss = self.loss_function(logits, labels, self.vocab_size, **loss_kwargs)
 
         if not return_dict: