Spaces:
Runtime error
Runtime error
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
|