Skip to content

Commit

Permalink
v0.1.1
Browse files Browse the repository at this point in the history
  • Loading branch information
v-zetang committed Mar 9, 2023
1 parent 6457d39 commit 156bb18
Showing 1 changed file with 20 additions and 0 deletions.
20 changes: 20 additions & 0 deletions README.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down

0 comments on commit 156bb18

Please sign in to comment.