JoPmt commited on
Commit
29034c2
1 Parent(s): a2045eb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -15,6 +15,7 @@ pipe.vqvae.to(torch.float32)
15
  pipe = pipe.to("cpu")
16
  def plex(prompt):
17
  prompt = prompt
 
18
  image = pipe(prompt, generator=generator)
19
  for a, imze in enumerate(image["images"]):
20
  apol.append(imze)
 
15
  pipe = pipe.to("cpu")
16
  def plex(prompt):
17
  prompt = prompt
18
+ generator = torch.Generator(device="cpu").manual_seed(random.randint(1, 4836923))
19
  image = pipe(prompt, generator=generator)
20
  for a, imze in enumerate(image["images"]):
21
  apol.append(imze)