Spaces:

black-forest-labs
/

FLUX.1-canny-dev

Running on Zero

multimodalart HF Staff commited on Nov 26, 2024

Commit

ec58781

verified ·

1 Parent(s): 34e644d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ from controlnet_aux import CannyDetector
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 2048
-pipe = FluxControlPipeline.from_pretrained("black-forest-labs/FLUX.1-Canny-dev", revision="refs/pr/1", torch_dtype=torch.bfloat16).to("cuda")
 processor = CannyDetector()
 @spaces.GPU

 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 2048
+pipe = FluxControlPipeline.from_pretrained("black-forest-labs/FLUX.1-Canny-dev", torch_dtype=torch.bfloat16).to("cuda")
 processor = CannyDetector()
 @spaces.GPU