Hugging Face's logo

Spaces:
SkalskiP
/
FLUX.1-inpaint-dev
Running on Zero

App Files Community
2
SkalskiP commited on
Commit
7189c81
•
1 Parent(s): b38c358

Florence-2 + SAM2 + FLUX.1

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -23,7 +23,8 @@ for taking it to the next level by enabling inpainting with the FLUX.
23
 
24
  MAX_SEED = np.iinfo(np.int32).max
25
  IMAGE_SIZE = 1024
26
- DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
27
 
28
  torch.autocast(device_type="cuda", dtype=torch.bfloat16).__enter__()
29
  if torch.cuda.get_device_properties(0).major >= 8:
 
23
 
24
  MAX_SEED = np.iinfo(np.int32).max
25
  IMAGE_SIZE = 1024
26
+ # DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
27
+ DEVICE = torch.device("cuda")
28
 
29
  torch.autocast(device_type="cuda", dtype=torch.bfloat16).__enter__()
30
  if torch.cuda.get_device_properties(0).major >= 8: