NikeZoldyck commited on
Commit
431e0cf
·
1 Parent(s): 90a5bb8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ pipe = StableDiffusionPipeline.from_pretrained(model_id,use_auth_token=token).to
27
 
28
  def infer(prompt,samples):
29
  with context(device):
30
- images = pipe(samples*[prompt],num_inference_steps=50, guidance_scale=7.5, height=400, width=400).images
31
  return images
32
 
33
 
 
27
 
28
  def infer(prompt,samples):
29
  with context(device):
30
+ images = pipe(samples*[prompt], guidance_scale=7.5).images
31
  return images
32
 
33