From 46107e3ecf129b155dac7de57edddbb1b1686113 Mon Sep 17 00:00:00 2001 From: "baiguan.yt" Date: Sat, 22 Oct 2022 20:31:59 +0800 Subject: [PATCH] [to #42322933]converting string to int to meet the input of face-image-generation Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/10489981 --- modelscope/pipelines/cv/face_image_generation_pipeline.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/modelscope/pipelines/cv/face_image_generation_pipeline.py b/modelscope/pipelines/cv/face_image_generation_pipeline.py index f00d639e..1b4e2e8a 100644 --- a/modelscope/pipelines/cv/face_image_generation_pipeline.py +++ b/modelscope/pipelines/cv/face_image_generation_pipeline.py @@ -61,6 +61,8 @@ class FaceImageGenerationPipeline(Pipeline): return input def forward(self, input: Dict[str, Any]) -> Dict[str, Any]: + if isinstance(input, str): + input = int(input) assert isinstance(input, int) torch.manual_seed(input) torch.cuda.manual_seed(input)