patrickvonplaten commited on
Commit
23a07e3
1 Parent(s): 6899ad2
Files changed (1) hide show
  1. image_upscaling.py +1 -0
image_upscaling.py CHANGED
@@ -22,6 +22,7 @@ IMAGE_UPSCALING_DESCRIPTION = (
22
  class ImageUpscalingTool(Tool):
23
  default_stable_diffusion_checkpoint = "stabilityai/sd-x2-latent-upscaler"
24
  description = IMAGE_UPSCALING_DESCRIPTION
 
25
  inputs = ['image']
26
  outputs = ['image']
27
 
 
22
  class ImageUpscalingTool(Tool):
23
  default_stable_diffusion_checkpoint = "stabilityai/sd-x2-latent-upscaler"
24
  description = IMAGE_UPSCALING_DESCRIPTION
25
+ name = "image_upscaler"
26
  inputs = ['image']
27
  outputs = ['image']
28