From 1c8f33967c83fad986a9e5401a468b2a7f7ccc0c Mon Sep 17 00:00:00 2001
From: Aryan <aryan@huggingface.co>
Date: Mon, 23 Dec 2024 14:59:05 +0100
Subject: [PATCH] add note about grad ckpt in readme

---
 README.md | 1 +
 1 file changed, 1 insertion(+)

diff --git a/README.md b/README.md
index 9c83a02..60dfc7f 100644
--- a/README.md
+++ b/README.md
@@ -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