Update modeling_gigarembed.py

modeling_gigarembed.py

@@ -360,7 +360,7 @@ class GigarEmbedModel(PreTrainedModel):
             # Mask out the instruction tokens for mean-pooling
             attention_mask[:, :instruction_lens] = 0
         features: GigarEmbedFeatures = {
-            'input_ids': batch_dict.get['input_ids'],
+            'input_ids': batch_dict['input_ids'],
             'attention_mask': batch_dict['attention_mask'],
             'pool_mask': attention_mask,
         }