ai-forever commited on
Commit
2b75c7d
·
verified ·
1 Parent(s): 0a89e97

delete unimported generator from pipe

Browse files
Files changed (1) hide show
  1. src/kandinsky.py +2 -2
src/kandinsky.py CHANGED
@@ -45,8 +45,8 @@ def generate_image(prompt, width, height, guidance_scale=3, num_inference_steps=
45
  guidance_scale = guidance_scale,
46
  num_inference_steps = num_inference_steps,
47
  width = width,
48
- height = height,
49
- generator = generator
50
  ).images[0]
51
 
52
  return image
 
45
  guidance_scale = guidance_scale,
46
  num_inference_steps = num_inference_steps,
47
  width = width,
48
+ height = height
49
+ # generator = generator
50
  ).images[0]
51
 
52
  return image