add note about grad ckpt in readme

This commit is contained in:
Aryan
2024-12-23 14:59:05 +01:00
parent 8c0c28c971
commit 1c8f33967c
+1
View File
@@ -386,6 +386,7 @@ If you would like to use a custom dataset, refer to the dataset preparation guid
> [!NOTE]
> To lower memory requirements:
> - Pass `--precompute_conditions` when launching training.
> - Pass `--gradient_checkpointing` when launching training.
> - Do not perform validation/testing. This saves a significant amount of memory, which can be used to focus solely on training if you're on smaller VRAM GPUs.
## Memory requirements