AlphaQuark commited on
Commit
6fdd551
·
verified ·
1 Parent(s): b4d543e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -14,6 +14,7 @@ pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-refine
14
  pipe = pipe.to(device)
15
 
16
  def infer(prompt, source_image, negative_prompt, guide, steps, seed, Strength):
 
17
  generator = torch.Generator(device).manual_seed(seed)
18
  if not isinstance(source_image, Image.Image):
19
  source_image = Image.open(source_image).convert("RGB")
 
14
  pipe = pipe.to(device)
15
 
16
  def infer(prompt, source_image, negative_prompt, guide, steps, seed, Strength):
17
+ seed = int(seed)
18
  generator = torch.Generator(device).manual_seed(seed)
19
  if not isinstance(source_image, Image.Image):
20
  source_image = Image.open(source_image).convert("RGB")