Spaces:
Runtime error
Runtime error
Commit
•
0d53d7d
1
Parent(s):
10af042
Update app.py
Browse files
app.py
CHANGED
@@ -12,7 +12,7 @@ from image_gen_aux import DepthPreprocessor
|
|
12 |
MAX_SEED = np.iinfo(np.int32).max
|
13 |
MAX_IMAGE_SIZE = 2048
|
14 |
|
15 |
-
pipe = FluxControlPipeline.from_pretrained("black-forest-labs/FLUX.1-Depth-dev",
|
16 |
processor = DepthPreprocessor.from_pretrained("LiheYoung/depth-anything-large-hf")
|
17 |
|
18 |
@spaces.GPU
|
|
|
12 |
MAX_SEED = np.iinfo(np.int32).max
|
13 |
MAX_IMAGE_SIZE = 2048
|
14 |
|
15 |
+
pipe = FluxControlPipeline.from_pretrained("black-forest-labs/FLUX.1-Depth-dev", torch_dtype=torch.bfloat16).to("cuda")
|
16 |
processor = DepthPreprocessor.from_pretrained("LiheYoung/depth-anything-large-hf")
|
17 |
|
18 |
@spaces.GPU
|