Prgckwb commited on
Commit
6850d81
1 Parent(s): 90deeeb

:tada: change some process

Browse files
Files changed (1) hide show
  1. src/inference.py +1 -1
src/inference.py CHANGED
@@ -106,7 +106,7 @@ def inference(
106
  height=height,
107
  guidance_scale=guidance_scale,
108
  num_inference_steps=num_inference_steps,
109
- num_images=num_images,
110
  generator=generator,
111
  ).images
112
 
 
106
  height=height,
107
  guidance_scale=guidance_scale,
108
  num_inference_steps=num_inference_steps,
109
+ num_images_per_prompt=num_images,
110
  generator=generator,
111
  ).images
112