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"
|