aknoerig commited on
Commit
681c549
·
1 Parent(s): fb49ce5

Currently the seed is ignored, as it is supplied as an unknown parameter (`seed`). Instead, it should be supplied as a generator.

Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -42,12 +42,15 @@ def predict(input_image, prompt, negative_prompt, steps, num_samples, scale, see
42
  init_image = input_image.convert("RGB")
43
  image = pad_image(init_image) # resize to integer multiple of 32
44
  image = image.resize((512, 512))
 
 
 
45
  result = dept2img(
46
  image=image,
47
  prompt=prompt,
48
  negative_prompt=negative_prompt,
49
  depth_image=depth,
50
- seed=seed,
51
  strength=strength,
52
  num_inference_steps=steps,
53
  guidance_scale=scale,
 
42
  init_image = input_image.convert("RGB")
43
  image = pad_image(init_image) # resize to integer multiple of 32
44
  image = image.resize((512, 512))
45
+ generator = None
46
+ if seed is not None:
47
+ generator = torch.Generator(device=device).manual_seed(seed)
48
  result = dept2img(
49
  image=image,
50
  prompt=prompt,
51
  negative_prompt=negative_prompt,
52
  depth_image=depth,
53
+ generator=generator,
54
  strength=strength,
55
  num_inference_steps=steps,
56
  guidance_scale=scale,