Skip to content

Commit 70cb0f0

Browse files
authored
Update modeling.py
1 parent 6ae1ddd commit 70cb0f0

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

FlagEmbedding/visual/modeling.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -301,13 +301,13 @@ def forward(self, mm_it_query=None, image_candidate=None, text_candidate=None, t
301301
### for stage-2 training
302302
if task_type == "edit_image":
303303
mm_query_reps = self.encode_mm(mm_it_query[0], mm_it_query[1])
304-
image_candi_reps = self.encode_image(image_candidate) #输入的是token序列
304+
image_candi_reps = self.encode_image(image_candidate)
305305
query_reps = mm_query_reps
306306
candi_reps = image_candi_reps
307307

308308
elif task_type == "t2it":
309309
text_query_reps = self.encode_text(text_query)
310-
mmit_candi_reps = self.encode_mm(mm_it_candidate[0], mm_it_candidate[1]) #输入的是token序列
310+
mmit_candi_reps = self.encode_mm(mm_it_candidate[0], mm_it_candidate[1])
311311
query_reps = text_query_reps
312312
candi_reps = mmit_candi_reps
313313

0 commit comments

Comments
 (0)