ImageGenerationModel

A loaded text-to-image model.

default_height (property)

default_width (property)

generate

generate(self, prompt: 'str', **options: 'Any') -> 'Any'

Generate images; returns an object with images (one [H, W, 3] tensor each) and seed. Options: negative_prompt, width, height, num_inference_steps, guidance_scale, seed, num_images, extra (a dict of family knobs).