Hugging Face's logo

Spaces:
huggingface-meta
/
FLUX.1-inpaint-dev2
Runtime error

App Files Community
SkalskiP commited on
Commit
4796c10
·
1 Parent(s): 902e7a4

Revert "Florence-2 + SAM2 + FLUX.1"

Browse files

This reverts commit 7189c81fddea1b41227e06bb0d5f33ddefc0024a.

Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -23,8 +23,7 @@ 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
- 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:
 
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: