INNER CODE UNIT · Python

update_workflow

fofr/cog-face-to-sticker · predict.py:56

    def update_workflow(self, workflow, **kwargs):
        loader = workflow["2"]["inputs"]
        sampler = workflow["4"]["inputs"]
        load_image = workflow["22"]["inputs"]
        upscaler = workflow["11"]["inputs"]
        instant_id = workflow["41"]["inputs"]
        ip_adapter = workflow["43"]["inputs"]

        load_image["image"] = kwargs["filename"]

        loader["cfg"] = kwargs["prompt_strength"]
        loader[
            "positive"
        ] = f"Sticker, {kwargs['prompt']}, cel shaded, svg, vector art, sharp"
        loader[
            "negative"
        ] = f"photo, photography, nsfw, nude, ugly, broken, watermark, oversaturated, soft {kwargs['negative_prompt']}"
        loader["empty_latent_width"] = kwargs["width"]

View source record →