alfredplpl
commited on
Commit
·
5a322c5
1
Parent(s):
db86cd0
Update app.py
Browse files
app.py
CHANGED
@@ -162,7 +162,7 @@ def txt_to_img(prompt, neg_prompt, guidance, steps, width, height, generator,sup
|
|
162 |
pipe=pipe.to("cuda")
|
163 |
upscaler=upscaler.to("cpu")
|
164 |
with torch.no_grad():
|
165 |
-
result =
|
166 |
else:
|
167 |
result = pipe(
|
168 |
prompt,
|
|
|
162 |
pipe=pipe.to("cuda")
|
163 |
upscaler=upscaler.to("cpu")
|
164 |
with torch.no_grad():
|
165 |
+
result = pipe.decode_latents(latents)
|
166 |
else:
|
167 |
result = pipe(
|
168 |
prompt,
|