Update app.py

app.py

@@ -10,7 +10,7 @@ import os, random, gc
 import numpy as np
 from accelerate import Accelerator
 accelerator = Accelerator(cpu=True)
-pipe = accelerator.prepare(AmusedPipeline.from_pretrained("amused/amused-512", variant="fp32", torch_dtype=torch.float32, use_safetensors=True))
+pipe = accelerator.prepare(AmusedPipeline.from_pretrained("amused/amused-512", variant=None, torch_dtype=torch.float32, use_safetensors=True))
 pipe.vqvae.to(torch.float32)
 pipe = pipe.to("cpu")
 def plex(prompt):