m-ric HF staff commited on
Commit
7ac1911
·
verified ·
1 Parent(s): 47ec38a

Update text_to_video.py

Browse files
Files changed (1) hide show
  1. text_to_video.py +2 -2
text_to_video.py CHANGED
@@ -19,8 +19,8 @@ class TextToVideoTool(Tool):
19
  default_checkpoint = "damo-vilab/text-to-video-ms-1.7b"
20
  description = TEXT_TO_VIDEO_DESCRIPTION
21
  name = "video_generator"
22
- inputs = {"prompt": {"type": str, "description": "contains the image description"}}
23
- output_type = Any
24
 
25
  def __init__(self, device=None, **hub_kwargs) -> None:
26
  if not is_accelerate_available():
 
19
  default_checkpoint = "damo-vilab/text-to-video-ms-1.7b"
20
  description = TEXT_TO_VIDEO_DESCRIPTION
21
  name = "video_generator"
22
+ inputs = {"prompt": {"type": "string", "description": "contains the image description"}}
23
+ output_type = "any"
24
 
25
  def __init__(self, device=None, **hub_kwargs) -> None:
26
  if not is_accelerate_available():