Spaces:

black-forest-labs
/

FLUX.1-Depth-dev

Running on Zero

multimodalart HF Staff commited on Nov 26, 2024

Commit

0d53d7d

verified ·

1 Parent(s): 10af042

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ from image_gen_aux import DepthPreprocessor
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 2048
-pipe = FluxControlPipeline.from_pretrained("black-forest-labs/FLUX.1-Depth-dev", revision="refs/pr/1", torch_dtype=torch.bfloat16).to("cuda")
 processor = DepthPreprocessor.from_pretrained("LiheYoung/depth-anything-large-hf")
 @spaces.GPU

 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 2048
+pipe = FluxControlPipeline.from_pretrained("black-forest-labs/FLUX.1-Depth-dev", torch_dtype=torch.bfloat16).to("cuda")
 processor = DepthPreprocessor.from_pretrained("LiheYoung/depth-anything-large-hf")
 @spaces.GPU