Spaces:
vilarin
/
Running on Zero

vilarin commited on
Commit
40c3357
·
verified ·
1 Parent(s): 508b41e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -106,6 +106,7 @@ def generate_image(
106
  num_inference_steps=steps,
107
  strength=strength,
108
  generator = generator,
 
109
  )
110
  else:
111
  image = pipe(
@@ -116,6 +117,7 @@ def generate_image(
116
  guidance_scale=scale,
117
  num_inference_steps=steps,
118
  generator = generator,
 
119
  )
120
  return image.images[0]
121
 
 
106
  num_inference_steps=steps,
107
  strength=strength,
108
  generator = generator,
109
+ prompt_embeds = None,
110
  )
111
  else:
112
  image = pipe(
 
117
  guidance_scale=scale,
118
  num_inference_steps=steps,
119
  generator = generator,
120
+ prompt_embeds = None,
121
  )
122
  return image.images[0]
123