Update app.py

app.py

@@ -11,8 +11,7 @@ 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))
-pipe = pipe.unet.to(memory_format=torch.channels_last)
+pipe = accelerator.prepare(pipe.vqvae.to(torch.float32, memory_format=torch.channels_last))
 pipe = pipe.to("cpu")
 apol=[]
 def plex(prompt, fifth, twice):