Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -17,7 +17,7 @@ def predict(steps, seed, value, guidance):
|
|
| 17 |
for i in range(1,steps):
|
| 18 |
yield pipeline(
|
| 19 |
generator=generator,
|
| 20 |
-
condition=torch.tensor([value, device=device
|
| 21 |
guidance=guidance,
|
| 22 |
num_inference_steps=steps
|
| 23 |
).images[0]
|
|
|
|
| 17 |
for i in range(1,steps):
|
| 18 |
yield pipeline(
|
| 19 |
generator=generator,
|
| 20 |
+
condition=torch.tensor([value], device=device),
|
| 21 |
guidance=guidance,
|
| 22 |
num_inference_steps=steps
|
| 23 |
).images[0]
|