Update app.py

app.py

@@ -11,14 +11,18 @@ import numpy as np
 from accelerate import Accelerator
 accelerator = Accelerator(cpu=True)
 pipe = accelerator.prepare(AmusedPipeline.from_pretrained("amused/amused-512", variant=None, torch_dtype=torch.float32, use_safetensors=True))
-pipe = accelerator.prepare(pipe.vqvae.to(torch.float32, memory_format=torch.channels_last))
+pipe = accelerator.prepare(pipe.vqvae.to(torch.float32))
 pipe = pipe.to("cpu")
 apol=[]
 def plex(prompt, fifth, twice):
     gc.collect()
     apol=[]
     prompt = prompt
-    image = pipe([prompt]*fifth, num_inference_steps=twice)
+    nm = random.randint(1, 4836928)
+    while nm % 32 != 0:
+        nm = random.randint(1, 4836928)
+    generator = torch.Generator(device="cpu").manual_seed(nm)
+    image = pipe([prompt]*fifth, generator=generator,num_inference_steps=twice)
     for a, imze in enumerate(image["images"]):
         apol.append(imze)
     return apol