Revert "debug"

This reverts commit 33f35f3a24399b1d7dadca5674ca010b3264272c.

app.py

@@ -24,12 +24,12 @@ for taking it to the next level by enabling inpainting with the FLUX.
 MAX_SEED = np.iinfo(np.int32).max
 IMAGE_SIZE = 1024
 # DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-DEVICE = torch.device("cpu")
+DEVICE = torch.device("cuda")
 
-# torch.autocast(device_type="cuda", dtype=torch.bfloat16).__enter__()
-# if torch.cuda.get_device_properties(0).major >= 8:
-#     torch.backends.cuda.matmul.allow_tf32 = True
-#     torch.backends.cudnn.allow_tf32 = True
+torch.autocast(device_type="cuda", dtype=torch.bfloat16).__enter__()
+if torch.cuda.get_device_properties(0).major >= 8:
+    torch.backends.cuda.matmul.allow_tf32 = True
+    torch.backends.cudnn.allow_tf32 = True
 
 FLORENCE_MODEL, FLORENCE_PROCESSOR = load_florence_model(device=DEVICE)
 SAM_IMAGE_MODEL = load_sam_image_model(device=DEVICE)
@@ -65,7 +65,7 @@ def is_image_empty(image: Image.Image) -> bool:
 
 @spaces.GPU()
 @torch.inference_mode()
-# @torch.autocast(device_type="cuda", dtype=torch.bfloat16)
+@torch.autocast(device_type="cuda", dtype=torch.bfloat16)
 def process(
     input_image_editor: dict,
     inpainting_prompt_text: str,