v0.1.1

weiyunhelong · Mar 9, 2023 · 06b4336 · 06b4336
1 parent 490a63e
commit 06b4336
Showing 1 changed file with 15 additions and 15 deletions.
diff --git a/README.md b/README.md
@@ -38,21 +38,21 @@ python visual_chatgpt.py
 ## GPU memory usage
 Here we list the GPU memory usage of each visual foundation model, 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 |
+| Fundation Model        | Memory Usage (MB) |
+|------------------------|-------------------|
+ | ImageEditing           | 6667              |
+| ImageCaption           | 1755              |
+| T2I                    | 6677              |
+| canny2image            | 5540              |
+| line2image             | 6679              |
+| hed2image              | 6679              |
+| scribble2image         | 6679              |
+| pose2image             | 6681              |
+| BLIPVQA                | 2709              |
+| seg2image              | 5540              |
+| depth2image            | 6677              |
+| normal2image           | 3974              |
+| Pix2Pix                | 2795              |