Skip to content

Prompt image generation engine

PromptImageGenerationEngine

Bases: BaseImageGenerationEngine

Source code in griptape/engines/image/prompt_image_generation_engine.py 
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
@define
class PromptImageGenerationEngine(BaseImageGenerationEngine):
    def run(
        self,
        prompts: list[str],
        *args,
        negative_prompts: Optional[list[str]] = None,
        rulesets: Optional[list[Ruleset]] = None,
        negative_rulesets: Optional[list[Ruleset]] = None,
        **kwargs,
    ) -> ImageArtifact:
        prompts = self._ruleset_to_prompts(prompts, rulesets)
        negative_prompts = self._ruleset_to_prompts(negative_prompts, negative_rulesets)

        return self.image_generation_driver.run_text_to_image(prompts, negative_prompts=negative_prompts)

run(prompts, *args, negative_prompts=None, rulesets=None, negative_rulesets=None, **kwargs)

Source code in griptape/engines/image/prompt_image_generation_engine.py 
14
15
16
17
18
19
20
21
22
23
24
25
26
def run(
    self,
    prompts: list[str],
    *args,
    negative_prompts: Optional[list[str]] = None,
    rulesets: Optional[list[Ruleset]] = None,
    negative_rulesets: Optional[list[Ruleset]] = None,
    **kwargs,
) -> ImageArtifact:
    prompts = self._ruleset_to_prompts(prompts, rulesets)
    negative_prompts = self._ruleset_to_prompts(negative_prompts, negative_rulesets)

    return self.image_generation_driver.run_text_to_image(prompts, negative_prompts=negative_prompts)