Spaces:
Running
on
Zero
Running
on
Zero
Revert "Florence-2 + SAM2 + FLUX.1"
Browse filesThis reverts commit 7189c81fddea1b41227e06bb0d5f33ddefc0024a.
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 |
-
|
| 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:
|