Spaces:
Runtime error
Runtime error
Update opensora/serve/gradio_web_server.py
Browse files
opensora/serve/gradio_web_server.py
CHANGED
@@ -67,6 +67,7 @@ if __name__ == '__main__':
|
|
67 |
|
68 |
vae = getae_wrapper(args.ae)(args.model_path, subfolder="vae").to(device, dtype=torch.float16)
|
69 |
vae.vae.enable_tiling()
|
|
|
70 |
image_size = int(args.version.split('x')[1])
|
71 |
latent_size = (image_size // ae_stride_config[args.ae][1], image_size // ae_stride_config[args.ae][2])
|
72 |
vae.latent_size = latent_size
|
|
|
67 |
|
68 |
vae = getae_wrapper(args.ae)(args.model_path, subfolder="vae").to(device, dtype=torch.float16)
|
69 |
vae.vae.enable_tiling()
|
70 |
+
vae.vae.tile_sample_min_size = 128
|
71 |
image_size = int(args.version.split('x')[1])
|
72 |
latent_size = (image_size // ae_stride_config[args.ae][1], image_size // ae_stride_config[args.ae][2])
|
73 |
vae.latent_size = latent_size
|