readme mistake?

[lingondricka2]

In magic prompt usage:

generator = RandomPromptGenerator()
magic_generator = MagicPromptGenerator(
    generator,
    device=...,  # Torch device specifier (int, string, torch.device)
)

num_prompts = 5
generator.generate("I love {red|green|blue} roses", num_prompts)

Should be?

generator = RandomPromptGenerator()
magic_generator = MagicPromptGenerator(
    generator,
    device=...,  # Torch device specifier (int, string, torch.device)
)

num_prompts = 5
magic_generator.generate("I love {red|green|blue} roses", num_prompts)