diff --git a/README.md b/README.md index 072c19b..2efe67c 100644 --- a/README.md +++ b/README.md @@ -66,7 +66,7 @@ GPU memory required by different settings during training are given below. The f Hardware: Single A100 80GB GPU with CPU RAM above 64G | Model | Full Finetuning | PEFT-LoRA | PEFT-LoRA with Gradient Checkpoitning | -| --------- | ---- | ---- | +| --------- | ---- | ---- | ---- | | CompVis/stable-diffusion-v1-4 | 27.5GB GPU / 3.97GB CPU | 15.5GB GPU / 3.84GB CPU | 8.12GB GPU / 3.77GB CPU |