Update app.py
Browse files
app.py
CHANGED
|
@@ -182,7 +182,7 @@ def generate(prompt: str, seed: int, randomize: bool):
|
|
| 182 |
)
|
| 183 |
|
| 184 |
with torch.no_grad(), torch.autocast("cpu", dtype=torch.bfloat16):
|
| 185 |
-
lat = _state["model"].sample_images_with_grad(noise, ctx, {"num_steps":
|
| 186 |
pix = _state["vae"].detokenize(lat)
|
| 187 |
|
| 188 |
arr = (pix[0].float().clamp(0, 1) * 255).round().byte()
|
|
|
|
| 182 |
)
|
| 183 |
|
| 184 |
with torch.no_grad(), torch.autocast("cpu", dtype=torch.bfloat16):
|
| 185 |
+
lat = _state["model"].sample_images_with_grad(noise, ctx, {"num_steps": 2})
|
| 186 |
pix = _state["vae"].detokenize(lat)
|
| 187 |
|
| 188 |
arr = (pix[0].float().clamp(0, 1) * 255).round().byte()
|