enable zerogpu

app.py

@@ -216,7 +216,7 @@ if NEW_MODEL:
     model.eval()
     print(missing_keys, extra_keys)
     assert len(missing_keys) == 0
-    vae_state_dict = torch.load(vae_path)['state_dict']
+    vae_state_dict = torch.load(vae_path,map_location='cpu')['state_dict']
     autoencoder = vqvae.create_model(3, 3, opts.latent_dim).eval().requires_grad_(False)
     missing_keys, extra_keys = autoencoder.load_state_dict(vae_state_dict, strict=False)
     autoencoer = autoencoder.to(device)