23 lines
604 B
Python
23 lines
604 B
Python
from __future__ import annotations
|
|
|
|
from dynamicprompts.generators.promptgenerator import PromptGenerator
|
|
|
|
|
|
class FrozenPromptGenerator(PromptGenerator):
|
|
"""
|
|
Generates a prompt once and repeats that prompt as num_images times
|
|
"""
|
|
|
|
def __init__(self, prompt_generator: PromptGenerator):
|
|
self._generator = prompt_generator
|
|
|
|
def generate(
|
|
self,
|
|
template: str,
|
|
num_images: int | None = 1,
|
|
**kwargs,
|
|
) -> list[str]:
|
|
prompts = self._generator.generate(template, 1)
|
|
num_images = num_images or 1
|
|
return num_images * prompts
|