From 156bb18423143c523c63a86bcde096c94e5da68c Mon Sep 17 00:00:00 2001 From: v-zetang Date: Thu, 9 Mar 2023 19:01:41 +0800 Subject: [PATCH] v0.1.1 --- README.md | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) diff --git a/README.md b/README.md index d95c339e..a4da253e 100644 --- a/README.md +++ b/README.md @@ -35,6 +35,26 @@ mkdir ./image python visual_chatgpt.py ``` +## GPU memory usage +Here we list the GPU memory usage of each visual foundation models, one can modify ``self.tools`` with fewer visual foundation models to save your GPU memory: + +| Fundation Model | Memory Usage (MB) | +| ---- | ---- | +| ImageInfo | 2501 | +| T2I | 6677 | +| canny2image | 5540 | +| line2image | 6679 | +| hed2image | 6679 | +| scribble2image | 6679 | +| pose2image | 6681 | +| BLIPVQA | 2709 | +| seg2image | 5540 | +| depth2image | 6677 | +| normal2image | 3974 | +| StableDiffusionInpaint | 8415 | +| Pix2Pix | 2795 | + + ## Acknowledgement We appreciate the open source of the following projects: