akhaliq HF Staff commited on
Commit
c20ac6a
·
1 Parent(s): 360d457
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -38,7 +38,7 @@ def predict(dict, prompt=""):
38
  with autocast("cuda"):
39
  init_image = dict["image"].convert("RGB").resize((512, 512))
40
  mask = dict["mask"].convert("RGB").resize((512, 512))
41
- output = pipe(prompt = prompt, image=init_image, mask_image=mask,guidance_scale=7.5,generator = torch.Generator(device="cuda").manual_seed(0))
42
  return output.images[0]
43
 
44
 
 
38
  with autocast("cuda"):
39
  init_image = dict["image"].convert("RGB").resize((512, 512))
40
  mask = dict["mask"].convert("RGB").resize((512, 512))
41
+ output = pipe(prompt = prompt, image=init_image, mask_image=mask,guidance_scale=7.5)
42
  return output.images[0]
43
 
44