Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2025-12-16 08:17:45 +01:00
Code Issues Packages Projects Releases Wiki Activity

stable diffusion allow postprocess kwargs

Browse Source 
Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/12418063
* stable diffusion allow postprocess kwargs
This commit is contained in:
zhangzhicheng.zzc
2023-04-21 18:34:29 +08:00
committed by hemu
parent 37b3d04824
commit fc9822ba85
3 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
modelscope/pipelines/multi_modal/diffusers_wrapped/diffusers_pipeline.py
View File

@@ -39,10 +39,10 @@ class DiffusersPipeline(Pipeline):
self.models = [self.model]
self.has_multiple_models = len(self.models) > 1
def preprocess(self, inputs: Dict[str, Any]) -> Dict[str, Any]:
def preprocess(self, inputs: Dict[str, Any], **kwargs) -> Dict[str, Any]:
return inputs
def postprocess(self, inputs: Dict[str, Any]) -> Dict[str, Any]:
def postprocess(self, inputs: Dict[str, Any], **kwargs) -> Dict[str, Any]:
return inputs
def __call__(self, input: Union[Input, List[Input]], *args,

2
modelscope/pipelines/multi_modal/diffusers_wrapped/stable_diffusion/chinese_stable_diffusion_pipeline.py
View File

@@ -75,7 +75,7 @@ class ChineseStableDiffusionPipeline(DiffusersPipeline):
callback=inputs.get('callback'),
callback_steps=inputs.get('callback_steps', 1))
def postprocess(self, inputs: Dict[str, Any]) -> Dict[str, Any]:
def postprocess(self, inputs: Dict[str, Any], **kwargs) -> Dict[str, Any]:
images = []
for img in inputs.images:
if isinstance(img, Image.Image):

2
modelscope/pipelines/multi_modal/diffusers_wrapped/stable_diffusion/stable_diffusion_pipeline.py
View File

@@ -65,7 +65,7 @@ class StableDiffusionWrapperPipeline(DiffusersPipeline):
callback=inputs.get('callback'),
callback_steps=inputs.get('callback_steps', 1))
def postprocess(self, inputs: Dict[str, Any]) -> Dict[str, Any]:
def postprocess(self, inputs: Dict[str, Any], **kwargs) -> Dict[str, Any]:
images = []
for img in inputs.images:
if isinstance(img, Image.Image):