patrickvonplaten commited on
Commit
bd76182
1 Parent(s): 429465d

Update image_transformation.py

Browse files
Files changed (1) hide show
  1. image_transformation.py +1 -0
image_transformation.py CHANGED
@@ -32,6 +32,7 @@ class ControlNetTransformationTool(Tool):
32
  default_stable_diffusion_checkpoint = "runwayml/stable-diffusion-v1-5"
33
  default_controlnet_checkpoint = "lllyasviel/control_v11p_sd15_canny"
34
  description = IMAGE_TRANSFORMATION_DESCRIPTION
 
35
  inputs = ['image', 'text']
36
  outputs = ['image']
37
 
 
32
  default_stable_diffusion_checkpoint = "runwayml/stable-diffusion-v1-5"
33
  default_controlnet_checkpoint = "lllyasviel/control_v11p_sd15_canny"
34
  description = IMAGE_TRANSFORMATION_DESCRIPTION
35
+ name = "image_transformer"
36
  inputs = ['image', 'text']
37
  outputs = ['image']
38