small fix to main

This commit is contained in:
Fabio 2024-08-19 21:23:32 +02:00
parent 76f5f01d4c
commit d1b4557d2e

View File

@ -28,7 +28,7 @@ def main():
flux = Flux1("black-forest-labs/FLUX.1-schnell", max_sequence_length=args.max_sequence_length)
image = flux.generate_image(
seed=args.seed,
seed=seed,
prompt=args.prompt,
config=Config(
num_inference_steps=args.steps,