Skip to content

Commit f4c40fc

Browse files
committed
grammar
Signed-off-by: Can-Zhao <[email protected]>
1 parent 047fb28 commit f4c40fc

File tree

1 file changed

+2
-5
lines changed

1 file changed

+2
-5
lines changed

generation/maisi/README.md

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -84,12 +84,9 @@ When `autoencoder_sliding_window_infer_size` is equal to or larger than the late
8484

8585

8686
### Training GPU Memory Usage
87-
The VAE is trained on patches and can be trained using a 16G GPU if the patch size is set to a small value, such as [64, 64, 64].
88-
Users can adjust the patch size to fit the available GPU memory.
89-
For the released model, we initially trained the autoencoder on a 16G V100 GPU with a small patch size of [64, 64, 64], and then continued training on a 32G V100 GPU with a larger patch size of [128, 128, 128].
87+
The VAE is trained on patches and can be trained using a 16G GPU if the patch size is set to a small value, such as [64, 64, 64]. Users can adjust the patch size to fit the available GPU memory. For the released model, we initially trained the autoencoder on a 16G V100 GPU with a small patch size of [64, 64, 64], and then continued training on a 32G V100 GPU with a larger patch size of [128, 128, 128].
9088

91-
The DM and ControlNet are trained on whole images rather than patches.
92-
The GPU memory usage during training depends on the size of the input images.
89+
The DM and ControlNet are trained on whole images rather than patches. The GPU memory usage during training depends on the size of the input images.
9390

9491
| image size | latent size | Peak Memory |
9592
|--------------|:------------- |:-----------:|

0 commit comments

Comments
 (0)