Spaces:
Runtime error
Runtime error
johnowhitaker
commited on
Commit
•
fb21251
1
Parent(s):
d7a843d
Update app.py
Browse files
app.py
CHANGED
@@ -30,7 +30,7 @@ def generate(color, guidance_loss_scale):
|
|
30 |
target_color = ImageColor.getcolor(color, "RGB") # Target color as RGB
|
31 |
target_color = [a/255 for a in target_color] # Rescale from (0, 255) to (0, 1)
|
32 |
x = torch.randn(1, 3, 256, 256).to(device)
|
33 |
-
for i, t in
|
34 |
model_input = scheduler.scale_model_input(x, t)
|
35 |
with torch.no_grad():
|
36 |
noise_pred = image_pipe.unet(model_input, t)["sample"]
|
|
|
30 |
target_color = ImageColor.getcolor(color, "RGB") # Target color as RGB
|
31 |
target_color = [a/255 for a in target_color] # Rescale from (0, 255) to (0, 1)
|
32 |
x = torch.randn(1, 3, 256, 256).to(device)
|
33 |
+
for i, t in enumerate(scheduler.timesteps):
|
34 |
model_input = scheduler.scale_model_input(x, t)
|
35 |
with torch.no_grad():
|
36 |
noise_pred = image_pipe.unet(model_input, t)["sample"]
|