Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -25,11 +25,11 @@ def split_image(input_image, num_splits=8):
|
|
25 |
output_images.append(input_image.crop(box))
|
26 |
return output_images
|
27 |
|
|
|
28 |
pipe = FluxPipeline.from_pretrained(
|
29 |
"black-forest-labs/FLUX.1-dev",
|
30 |
torch_dtype=torch_dtype,
|
31 |
-
use_safetensors=True
|
32 |
-
variant="fp16"
|
33 |
).to(device)
|
34 |
|
35 |
MAX_SEED = np.iinfo(np.int32).max
|
|
|
25 |
output_images.append(input_image.crop(box))
|
26 |
return output_images
|
27 |
|
28 |
+
# 파이프라인 초기화 수정
|
29 |
pipe = FluxPipeline.from_pretrained(
|
30 |
"black-forest-labs/FLUX.1-dev",
|
31 |
torch_dtype=torch_dtype,
|
32 |
+
use_safetensors=True
|
|
|
33 |
).to(device)
|
34 |
|
35 |
MAX_SEED = np.iinfo(np.int32).max
|