JoPmt commited on
Commit
cf9ba41
1 Parent(s): 99c67f4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -13,7 +13,10 @@ apol=[]
13
  pipe = accelerator.prepare(AutoPipelineForText2Image.from_pretrained("openskyml/overall-v1", torch_dtype=torch.float32, variant=None, use_safetensors=False, safety_checker=None))
14
  pipe = accelerator.prepare(pipe.to("cpu"))
15
  def plex(prompt,neg_prompt):
16
- generator = torch.Generator(device="cpu").manual_seed(random.randint(1, 4836923))
 
 
 
17
  image = pipe(prompt=prompt, negative_prompt=neg_prompt, generator=generator, num_inference_steps=15)
18
  for a, imze in enumerate(image["images"]):
19
  apol.append(imze)
 
13
  pipe = accelerator.prepare(AutoPipelineForText2Image.from_pretrained("openskyml/overall-v1", torch_dtype=torch.float32, variant=None, use_safetensors=False, safety_checker=None))
14
  pipe = accelerator.prepare(pipe.to("cpu"))
15
  def plex(prompt,neg_prompt):
16
+ nm = random.randint(1, 4836928)
17
+ while nm % 32 != 0:
18
+ nm = random.randint(1, 4836928)
19
+ generator = torch.Generator(device="cpu").manual_seed(nm)
20
  image = pipe(prompt=prompt, negative_prompt=neg_prompt, generator=generator, num_inference_steps=15)
21
  for a, imze in enumerate(image["images"]):
22
  apol.append(imze)