IO
Browse files- image_transformation.py +2 -0
image_transformation.py
CHANGED
|
@@ -32,6 +32,8 @@ class ImageTransformationTool(Tool):
|
|
| 32 |
default_stable_diffusion_checkpoint = "runwayml/stable-diffusion-v1-5"
|
| 33 |
default_controlnet_checkpoint = "lllyasviel/sd-controlnet-canny"
|
| 34 |
description = IMAGE_TRANSFORMATION_DESCRIPTION
|
|
|
|
|
|
|
| 35 |
|
| 36 |
def __init__(self, device=None, controlnet=None, stable_diffusion=None, **hub_kwargs) -> None:
|
| 37 |
if not is_accelerate_available():
|
|
|
|
| 32 |
default_stable_diffusion_checkpoint = "runwayml/stable-diffusion-v1-5"
|
| 33 |
default_controlnet_checkpoint = "lllyasviel/sd-controlnet-canny"
|
| 34 |
description = IMAGE_TRANSFORMATION_DESCRIPTION
|
| 35 |
+
inputs = ['image', 'text']
|
| 36 |
+
outputs = ['image']
|
| 37 |
|
| 38 |
def __init__(self, device=None, controlnet=None, stable_diffusion=None, **hub_kwargs) -> None:
|
| 39 |
if not is_accelerate_available():
|