Commit
•
e6c5bbf
1
Parent(s):
171d4cc
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
|