Spaces:
Runtime error
Runtime error
Commit
•
c99dfbf
1
Parent(s):
c70f6b6
Update app.py
Browse files
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/
|
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)):
|