Spaces:
Sleeping
Sleeping
francescoKrnl
commited on
Commit
•
bbd66f8
1
Parent(s):
54d7710
changed r= 0.6
Browse files
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.
|
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 |
|