johnowhitaker commited on
Commit
fb21251
1 Parent(s): d7a843d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 tqdm(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"]
 
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"]