Update app.py
Browse filestry to get rid of error
app.py
CHANGED
@@ -26,13 +26,13 @@ control_img = 'myimage.jpg'
|
|
26 |
|
27 |
examples = [["a yellow dog in grass", "lowres, two heads, bad muzzle, bad anatomy, missing ears, missing paws", "example1.jpg"]]
|
28 |
|
29 |
-
default_example = examples[0]
|
30 |
|
31 |
controlnet, controlnet_params = FlaxControlNetModel.from_pretrained(
|
32 |
"JFoz/dog-cat-pose", dtype=jnp.bfloat16
|
33 |
)
|
34 |
pipe, params = FlaxStableDiffusionControlNetPipeline.from_pretrained(
|
35 |
-
"runwayml/stable-diffusion-v1-5", controlnet=controlnet, revision="flax", dtype=jnp.bfloat16
|
36 |
)
|
37 |
|
38 |
def infer(prompts, negative_prompts, image):
|
|
|
26 |
|
27 |
examples = [["a yellow dog in grass", "lowres, two heads, bad muzzle, bad anatomy, missing ears, missing paws", "example1.jpg"]]
|
28 |
|
29 |
+
#default_example = examples[0]
|
30 |
|
31 |
controlnet, controlnet_params = FlaxControlNetModel.from_pretrained(
|
32 |
"JFoz/dog-cat-pose", dtype=jnp.bfloat16
|
33 |
)
|
34 |
pipe, params = FlaxStableDiffusionControlNetPipeline.from_pretrained(
|
35 |
+
"runwayml/stable-diffusion-v1-5", controlnet=controlnet, revision="flax", dtype=jnp.bfloat16, safety_checker=None,
|
36 |
)
|
37 |
|
38 |
def infer(prompts, negative_prompts, image):
|