kfahn commited on
Commit
936aa8d
1 Parent(s): 1bb6ad6

Update app.py

Browse files

try to get rid of error

Files changed (1) hide show
  1. app.py +2 -2
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):