change cuda

app.py

@@ -1005,6 +1005,7 @@ def handle_upload(files, chat_state, img_list):
         
 #     return gr.update(value=[]), gr.update(value=None, interactive=True), gr.update(placeholder='Please upload your image first', interactive=False),gr.update(value="Upload & Start Chat", interactive=True), gr.update(interactive=False), chat_state, img_list
 # chatbot, image, text_input, upload_button, submit_button, chat_state, img_list
+
 def encode_img(img_list, modalities):
     instruction, LR_image_list, HR_image_list, modalities = chat.encode_img(img_list, modalities)
     return instruction, LR_image_list, HR_image_list, modalities
@@ -1211,7 +1212,7 @@ def upload_img(chatbot,t1c_ax_file, t1_file, t2_file, fla_file, text_input, chat
     img_list = []
     llm_message = chat.upload_img(files, chat_state, img_list)
     
-    instruction, LR_image_list, HR_image_list, modalities = encode_img(img_list, modalities)
+    instruction, LR_image_list.to('cuda'), HR_image_list.to('cuda'), modalities = encode_img(img_list, modalities)
     # instruction, LR_image_list, HR_image_list, modalities = chat.encode_img(img_list, modalities)
     return chatbot, chat_state, instruction, LR_image_list, HR_image_list, modalities, gr.update(interactive=False),gr.update(interactive=False)
 

minigpt4/conversation/conversation.py

@@ -298,7 +298,7 @@ def images_process(image_list,images,modalities=None):#get_item
                 
             # aug_probability = np.random.rand() 
             
-            image_sequence = image_sequence.squeeze().cuda()  # torch.Size([32, 630, 637])
+            image_sequence = image_sequence.squeeze()  # torch.Size([32, 630, 637])
                         
             image_sequence_LR = our_data_resize(image_sequence)  # [32, 256, 256] -> [32, 224, 224]
             
@@ -334,10 +334,10 @@ def images_process(image_list,images,modalities=None):#get_item
 
 
             
-    LR_image_list = torch.stack(LR_image_list,dim=0).cuda()  # torch.Size([7, 32, 3, 224, 224])
+    LR_image_list = torch.stack(LR_image_list,dim=0)  # torch.Size([7, 32, 3, 224, 224])
 
     if HR_resolution==True and HR_image_list!=[]:
-        HR_image_list = torch.stack(HR_image_list, dim=0).cuda()  # torch.Size([7, 128, 3, 224, 224])
+        HR_image_list = torch.stack(HR_image_list, dim=0)  # torch.Size([7, 128, 3, 224, 224])
     else:
         HR_resolution=False
         HR_image_list = []