Update opensora/serve/gradio_web_server.py

opensora/serve/gradio_web_server.py

@@ -65,9 +65,9 @@ if __name__ == '__main__':
     # Load model:
     transformer_model = LatteT2V.from_pretrained(args.model_path, subfolder=args.version, torch_dtype=torch.float16).to(device)
 
-    vae = getae_wrapper(args.ae)(args.model_path, subfolder="vae").to(device, dtype=torch.float16)
+    vae = getae_wrapper(args.ae)(args.model_path, subfolder="vae").to(device)
+    vae = vae.half()
     vae.vae.enable_tiling()
-    vae.vae.tile_sample_min_size = 128
     image_size = int(args.version.split('x')[1])
     latent_size = (image_size // ae_stride_config[args.ae][1], image_size // ae_stride_config[args.ae][2])
     vae.latent_size = latent_size