ControlNet-with-Anything-v4

Runtime error

hysts HF staff commited on Mar 5, 2023

Commit

d1c233a

•

1 Parent(s): 8d8da79

Fix

Files changed (1) hide show

model.py CHANGED Viewed

@@ -119,6 +119,8 @@ class Model:
         guidance_scale: float,
         seed: int,
     ) -> list[PIL.Image.Image]:
         generator = torch.Generator().manual_seed(seed)
         return self.pipe(prompt=prompt,
                          negative_prompt=negative_prompt,

         guidance_scale: float,
         seed: int,
     ) -> list[PIL.Image.Image]:
+        if seed == -1:
+            seed = np.random.randint(0, np.iinfo(np.int64).max)
         generator = torch.Generator().manual_seed(seed)
         return self.pipe(prompt=prompt,
                          negative_prompt=negative_prompt,