Fix seed (#75)

- Fix seed (681c5498977e01d6825197e5ca7968010fe4f8eb)
- Merge branch 'main' into pr/75 (1553a94271ada2b1e679651225c4666a8be61312)


Co-authored-by: André Knörig <[email protected]>

app.py

@@ -42,12 +42,15 @@ def predict(input_image, prompt, negative_prompt, steps, num_samples, scale, see
     init_image = input_image.convert("RGB")
     image = pad_image(init_image)  # resize to integer multiple of 32
     image = image.resize((512, 512))
+    generator = None
+    if seed is not None:
+        generator = torch.Generator(device=device).manual_seed(seed)
     result = dept2img(
         image=image,
         prompt=prompt,
         negative_prompt=negative_prompt,
+        generator=generator,
         depth_map=depth,
-        seed=seed,
         strength=strength,
         num_inference_steps=steps,
         guidance_scale=scale,