Skip to content

Commit 3618cc8

Browse files
committed
update readme
Signed-off-by: Can-Zhao <volcanofly@gmail.com>
1 parent ace8223 commit 3618cc8

File tree

1 file changed

+1
-0
lines changed
  • models/brats_mri_generative_diffusion/docs

1 file changed

+1
-0
lines changed

models/brats_mri_generative_diffusion/docs/README.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,7 @@ We need to install the required packages.
4141
```
4242
pip install git+https://github.com/Project-MONAI/GenerativeModels.git
4343
pip install lpips
44+
pip install xformers
4445
```
4546

4647
If you have a GPU with less than 32G of memory, you may need to decrease the batch size when training. To do so, modify the `train_batch_size` parameter in the [configs/train_autoencoder.json](../configs/train_autoencoder.json) and [configs/train_diffusion.json](../configs/train_diffusion.json) configuration files.

0 commit comments

Comments
 (0)