multimodalart HF staff commited on
Commit
c99dfbf
1 Parent(s): c70f6b6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ from diffusers.utils import load_image
11
  MAX_SEED = np.iinfo(np.int32).max
12
  MAX_IMAGE_SIZE = 2048
13
 
14
- pipe = FluxControlPipeline.from_pretrained("black-forest-labs/FLUX.1-Depth-dev", revision="refs/pr/8", 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)):
 
11
  MAX_SEED = np.iinfo(np.int32).max
12
  MAX_IMAGE_SIZE = 2048
13
 
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)):