FLUX.1-inpaint-dev2

Runtime error

SkalskiP commited on Aug 13, 2024

Commit

1bab1e8

1 Parent(s): d888127

debug

Files changed (1) hide show

app.py CHANGED Viewed

@@ -23,8 +23,8 @@ 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")
 torch.autocast(device_type="cuda", dtype=torch.bfloat16).__enter__()
 if torch.cuda.get_device_properties(0).major >= 8:

 MAX_SEED = np.iinfo(np.int32).max
 IMAGE_SIZE = 1024
+# DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+DEVICE = torch.device("cuda")
 torch.autocast(device_type="cuda", dtype=torch.bfloat16).__enter__()
 if torch.cuda.get_device_properties(0).major >= 8: