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(
|
| 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),
|
| 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 |
|