jcccy commited on
Commit
7db9607
·
1 Parent(s): 02c8781

change cuda

Browse files
minigpt4/conversation/conversation.py CHANGED
@@ -605,7 +605,7 @@ class Chat:
605
  confidence_list=[]
606
  image_index_list=find_same_view(modalities) #
607
  # breakpoint()
608
- for img_idx in range(min(5,len(image_index_list))):
609
  # images=[image_list_name[img_i] for img_i in image_index_list[img_idx]]
610
  # sub_modalities=[modalities[img_i] for img_i in image_index_list[img_idx]]
611
  # instruction,image_list,HR_image_list=images_process(images,v,sub_modalities)
@@ -721,7 +721,7 @@ class Chat:
721
  combinations = itertools.combinations([_ for _ in range(len(lr_img_list))], min(5,len(lr_img_list)))
722
  combinations=list(combinations)
723
  random.shuffle(combinations)
724
- combinations=combinations[0:min(len(combinations),5)]#测CN4组合,但是random select 5个人或更少
725
 
726
 
727
 
 
605
  confidence_list=[]
606
  image_index_list=find_same_view(modalities) #
607
  # breakpoint()
608
+ for img_idx in range(min(3,len(image_index_list))):
609
  # images=[image_list_name[img_i] for img_i in image_index_list[img_idx]]
610
  # sub_modalities=[modalities[img_i] for img_i in image_index_list[img_idx]]
611
  # instruction,image_list,HR_image_list=images_process(images,v,sub_modalities)
 
721
  combinations = itertools.combinations([_ for _ in range(len(lr_img_list))], min(5,len(lr_img_list)))
722
  combinations=list(combinations)
723
  random.shuffle(combinations)
724
+ combinations=combinations[0:min(len(combinations),3)]#测CN4组合,但是random select 5个人或更少
725
 
726
 
727