Update main.py

main.py

@@ -8,7 +8,12 @@ from pipeline_dedit_sd import DEditSDPipeline
 from utils import load_image, load_mask, load_mask_edit
 from utils_mask import process_mask_move_torch, process_mask_remove_torch, mask_union_torch, mask_substract_torch, create_outer_edge_mask_torch
 from utils_mask import check_mask_overlap_torch, check_cover_all_torch, visualize_mask_list, get_mask_difference_torch, save_mask_list_to_npys
-
+print(torch.__version__)
+if torch.cuda.is_available():
+    print("CUDA version:", torch.version.cuda)
+    print("GPU model name:", torch.cuda.get_device_name(0))
+else:
+    print('no cuda.')
 def run_main(
     name="example_tmp",
     name_2=None,