Spaces:
Running
Running
Update app.py
Browse files
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=
|
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
|