JoPmt commited on
Commit
52f2371
1 Parent(s): dbf39c7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ def plex(prompt,neg_prompt):
14
  apol=[]
15
  pipe = accelerator.prepare(AutoPipelineForText2Image.from_pretrained("openskyml/overall-v1", torch_dtype=torch.float32, variant=None, use_safetensors=False, safety_checker=None))
16
  pipe = accelerator.prepare(pipe.to("cpu"))
17
- image = pipe(prompt=prompt, negative_prompt=neg_prompt,num_inference_steps=10)
18
  for a, imze in enumerate(image["images"]):
19
  apol.append(imze)
20
  return apol
 
14
  apol=[]
15
  pipe = accelerator.prepare(AutoPipelineForText2Image.from_pretrained("openskyml/overall-v1", torch_dtype=torch.float32, variant=None, use_safetensors=False, safety_checker=None))
16
  pipe = accelerator.prepare(pipe.to("cpu"))
17
+ image = pipe(prompt=prompt, negative_prompt=neg_prompt, generator=generator, num_inference_steps=15)
18
  for a, imze in enumerate(image["images"]):
19
  apol.append(imze)
20
  return apol