Skip to content

Commit

Permalink
prepare deprecation of is_vision_model
Browse files Browse the repository at this point in the history
  • Loading branch information
ManuelFay committed Sep 26, 2024
1 parent 1a53878 commit ac44961
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion colpali_engine/trainer/contrastive_trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@ class ContrastiveTrainer(Trainer):
def __init__(self, loss_func, is_vision_model, *args, **kwargs):
super().__init__(*args, **kwargs)
self.loss_func = loss_func
self.is_vision_model = is_vision_model
self.is_vision_model = is_vision_model # Unused argument, will be removed in 0.4.0

def compute_loss(self, model, inputs, return_outputs=False):
query_outputs = model(input_ids=inputs["query_input_ids"], attention_mask=inputs["query_attention_mask"])
Expand Down

0 comments on commit ac44961

Please sign in to comment.