Hugging Face's logo

Spaces:
black-forest-labs
/
FLUX.1-Depth-dev
Runtime error

App Files Community
multimodalart HF staff commited on
Commit
1d15b14
•
1 Parent(s): c99dfbf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -14,6 +14,7 @@ MAX_IMAGE_SIZE = 2048
14
  pipe = FluxControlPipeline.from_pretrained("black-forest-labs/FLUX.1-Depth-dev", revision="refs/pr/1", torch_dtype=torch.bfloat16).to("cuda")
15
  processor = DepthPreprocessor.from_pretrained("LiheYoung/depth-anything-large-hf")
16
 
 
17
  def infer(control_image, prompt, seed=42, randomize_seed=False, width=1024, height=1024, guidance_scale=3.5, num_inference_steps=28, progress=gr.Progress(track_tqdm=True)):
18
 
19
  if randomize_seed:
 
14
  pipe = FluxControlPipeline.from_pretrained("black-forest-labs/FLUX.1-Depth-dev", revision="refs/pr/1", torch_dtype=torch.bfloat16).to("cuda")
15
  processor = DepthPreprocessor.from_pretrained("LiheYoung/depth-anything-large-hf")
16
 
17
+ @spaces.GPU
18
  def infer(control_image, prompt, seed=42, randomize_seed=False, width=1024, height=1024, guidance_scale=3.5, num_inference_steps=28, progress=gr.Progress(track_tqdm=True)):
19
 
20
  if randomize_seed: