francescoKrnl commited on
Commit
bbd66f8
1 Parent(s): 54d7710

changed r= 0.6

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -58,7 +58,7 @@ def run(image, item_name, color_name):
58
  torch.manual_seed(42)
59
  B,C,H,W = c_t.shape
60
  noise = torch.randn((1,4,H//8, W//8), device=c_t.device)
61
- output_image = model(c_t, prompt, deterministic=False, r=0.4, noise_map=noise)
62
  output_pil = TF.to_pil_image(output_image[0].cpu()*0.5+0.5)
63
  return output_pil
64
 
 
58
  torch.manual_seed(42)
59
  B,C,H,W = c_t.shape
60
  noise = torch.randn((1,4,H//8, W//8), device=c_t.device)
61
+ output_image = model(c_t, prompt, deterministic=False, r=0.6, noise_map=noise)
62
  output_pil = TF.to_pil_image(output_image[0].cpu()*0.5+0.5)
63
  return output_pil
64