From 4e577cf6e37a4eed61999b80b20fcfb208406930 Mon Sep 17 00:00:00 2001 From: Sourab Mangrulkar <13534540+pacman100@users.noreply.github.com> Date: Tue, 3 Jan 2023 23:13:40 +0530 Subject: [PATCH] Update README.md --- README.md | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index 7c9acfe..41ff4ee 100644 --- a/README.md +++ b/README.md @@ -66,7 +66,7 @@ GPU memory and training times required by different settings during training are Hardware: Single A100 80GB GPU with CPU RAM above 64G | Model | Full Finetuning | PET-LoRA | -| --------- | ---- | ---- | ---- | +| --------- | ---- | ---- | | CompVis/stable-diffusion-v1-4 | 66.4GB GPU / 3.97GB CPU / 22 minutes | 15.5GB GPU / 3.84GB CPU / 5.5 minutes | @@ -182,6 +182,8 @@ Use 🤗 Accelerate for inferencing on consumer hardware with small resources. ### Example of PET model inference using 🤗 Accelerate's Big Model Inferencing capabilities +Example is provided in `~examples/causal_language_modeling/pet_lora_clm_accelerate_big_model_inference.ipynb`. + ## Models support matrix