multimodalart HF staff commited on
Commit
0d53d7d
1 Parent(s): 10af042

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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", revision="refs/pr/1", torch_dtype=torch.bfloat16).to("cuda")
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