Skip to content

Commit

Permalink
Update unconditional_image_generation.py
Browse files Browse the repository at this point in the history
  • Loading branch information
deepak-gowda-narayana authored Oct 30, 2024
1 parent 49437da commit 0f47121
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -109,7 +109,7 @@ def main():
kwargs_call = {"throughput_warmup_steps": args.throughput_warmup_steps}

pipeline = GaudiDDPMPipeline.from_pretrained(model_name, **kwargs)
output = pipeline(batch_size=args.batch_size, num_inference_steps=args.num_inference_steps)
output = pipeline(batch_size=args.batch_size, num_inference_steps=args.num_inference_steps, **kwargs_call)

if args.output_dir:
logger.info(f"Generating outputs to {args.output_dir}")
Expand Down

0 comments on commit 0f47121

Please sign in to comment.