Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -53,7 +53,7 @@ models =[
|
|
| 53 |
]
|
| 54 |
|
| 55 |
def plex(prompt,goof,modil):
|
| 56 |
-
pipe = accelerator.prepare(AutoPipelineForText2Image.from_pretrained(
|
| 57 |
pipe = accelerator.prepare(pipe.to("cpu"))
|
| 58 |
# prompt = "A fantasy landscape, Cinematic lighting"
|
| 59 |
# negative_prompt = "low quality, bad quality"
|
|
|
|
| 53 |
]
|
| 54 |
|
| 55 |
def plex(prompt,goof,modil):
|
| 56 |
+
pipe = accelerator.prepare(AutoPipelineForText2Image.from_pretrained(""+modil+"", torch_dtype=torch.float32))
|
| 57 |
pipe = accelerator.prepare(pipe.to("cpu"))
|
| 58 |
# prompt = "A fantasy landscape, Cinematic lighting"
|
| 59 |
# negative_prompt = "low quality, bad quality"
|