change cuda

minigpt4/conversation/conversation.py

@@ -675,11 +675,11 @@ class Chat:
                 # thread0.join()
                 # thread1.join()
                 
-                # try:
-                #     self.model_list[0].to('cuda')
-                #     answer = self.model_list[0].generate_step(lr_img_list_combination, confidence_instruction, hr_img_list_combination)
-                # except Exception as e:
-                #     print('model_0 error',e)
+                try:
+                    self.model_list[0].to('cuda')
+                    answer = self.model_list[0].generate_step(lr_img_list_combination, confidence_instruction, hr_img_list_combination)
+                except Exception as e:
+                    print('model_0 error',e)
 
                 try:
                     self.model_list[1].to('cuda')
@@ -778,11 +778,11 @@ class Chat:
                 # answer = answer[0]
                 # answer_new = answer_new[0]
                 
-                # try:
-                #     self.model_list[0].to('cuda')
-                #     answer = self.model_list[0].generate_step(lr_img_list_combination, confidence_instruction, hr_img_list_combination)
-                # except Exception as e:
-                #     print('model_0 error',e)
+                try:
+                    self.model_list[0].to('cuda')
+                    answer = self.model_list[0].generate_step(lr_img_list_combination, confidence_instruction, hr_img_list_combination)
+                except Exception as e:
+                    print('model_0 error',e)
                 
                 try:
                     self.model_list[1].to('cuda')
@@ -790,7 +790,7 @@ class Chat:
                 except Exception as e:
                     print('model_1 error',e)
                 
-                answer=answer_new
+                # answer=answer_new
                 output_text=answer
             
                 answer_list.append(answer_new)