visual-chatgpt icon indicating copy to clipboard operation
visual-chatgpt copied to clipboard

GPU Requirement

Open Hariharan2608 opened this issue 2 years ago • 6 comments

How many GB of GPU is required to run this project?

Hariharan2608 avatar Mar 11 '23 04:03 Hariharan2608

It depends on what models you need to use, check out this readme here https://github.com/microsoft/visual-chatgpt#gpu-memory-usage

K-tang-mkv avatar Mar 11 '23 04:03 K-tang-mkv

I already implement the Google Colab version and I only choose two models T2I and ImageCaption, everything is fine, you can try it on my Colab notebook, checkout this colab version

K-tang-mkv avatar Mar 11 '23 06:03 K-tang-mkv

You're better off using one of the forks/colab versions that people have got running. To run this project without modification requires 8x GPUs.

Let me quote myself from here:

I did get this running in the end, on 8x NVIDIA A100 40 Gb, but various bugs prevented it from fully working (for one, the masking for inpainting wasn't working, not sure if the fork uses/fixes this model?).

"Anything goes, so long as you have 70 GB of VRAM..."

Not quite, I tried running it on 8x NVIDIA Tesla V100, 16 Gb. But got OOM on one of the cards when trying to generate an image. I.e. the cards need to be big enough to run the models allocated to them.

...Looking forward to the multimodal APIs coming soon, as you say.

sradc avatar Mar 11 '23 07:03 sradc

I already implement the Google Colab version and I only choose two models T2I and ImageCaption, everything is fine, you can try it on my Colab notebook, checkout this colab version

兄弟你api key露出来了

tzcnbo avatar Mar 11 '23 10:03 tzcnbo

@tzcnbo 我靠,疏忽了,谢谢兄弟的提醒

K-tang-mkv avatar Mar 11 '23 10:03 K-tang-mkv

ניתן למחוק את ה- API שבחשבון ולייצר אותו מחדש, למקרה שזה יהיה נחוץ.

FoehnWorknet avatar Mar 11 '23 12:03 FoehnWorknet