small fix to main
This commit is contained in:
parent
76f5f01d4c
commit
d1b4557d2e
2
main.py
2
main.py
@ -28,7 +28,7 @@ def main():
|
|||||||
flux = Flux1("black-forest-labs/FLUX.1-schnell", max_sequence_length=args.max_sequence_length)
|
flux = Flux1("black-forest-labs/FLUX.1-schnell", max_sequence_length=args.max_sequence_length)
|
||||||
|
|
||||||
image = flux.generate_image(
|
image = flux.generate_image(
|
||||||
seed=args.seed,
|
seed=seed,
|
||||||
prompt=args.prompt,
|
prompt=args.prompt,
|
||||||
config=Config(
|
config=Config(
|
||||||
num_inference_steps=args.steps,
|
num_inference_steps=args.steps,
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user