JoPmt commited on
Commit
e6d2e23
·
1 Parent(s): f6e96b9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ from accelerate import Accelerator
9
  accelerator = Accelerator()
10
 
11
  def plex(prompt,neg_prompt):
12
- pipe = accelerator.prepare(AutoPipelineForText2Image.from_pretrained("openskyml/overall-v1", torch_dtype=torch.float32, variant=None, use_safetensors=False, safety_checker=False))
13
  pipe = accelerator.prepare(pipe.to("cpu"))
14
  image = pipe(prompt=prompt, negative_prompt=neg_prompt,num_inference_steps=10).images[0]
15
  return image
 
9
  accelerator = Accelerator()
10
 
11
  def plex(prompt,neg_prompt):
12
+ pipe = accelerator.prepare(AutoPipelineForText2Image.from_pretrained("openskyml/overall-v1", torch_dtype=torch.float32, variant=None, use_safetensors=False, safety_checker=None))
13
  pipe = accelerator.prepare(pipe.to("cpu"))
14
  image = pipe(prompt=prompt, negative_prompt=neg_prompt,num_inference_steps=10).images[0]
15
  return image