Spaces:
Running
on
Zero
Running
on
Zero
update
Browse files
app.py
CHANGED
|
@@ -16,7 +16,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
| 16 |
pipe = FluxPipeline.from_pretrained(
|
| 17 |
"black-forest-labs/FLUX.1-dev",
|
| 18 |
torch_dtype=torch.bfloat16
|
| 19 |
-
).to(
|
| 20 |
|
| 21 |
def split_image(input_image, num_splits=4):
|
| 22 |
# Create a list to store the output images
|
|
|
|
| 16 |
pipe = FluxPipeline.from_pretrained(
|
| 17 |
"black-forest-labs/FLUX.1-dev",
|
| 18 |
torch_dtype=torch.bfloat16
|
| 19 |
+
).to(device)
|
| 20 |
|
| 21 |
def split_image(input_image, num_splits=4):
|
| 22 |
# Create a list to store the output images
|