InternLM / InternLM/InternLM-XComposer

How much VRAM required for Share Captioner?

Open
#232 3 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
2.9k
Forks
175
PR merge metrics
No merged PRs in 30d

Description

Hi

1) How much VRAM is required for Share Captioner?

2) Is there a way to use Multiple GPUs for loading Share Captioner?

3) Are there Quantization methods (4bit, 8bit) available for Share Captioner?

-------------------------------------------------------------------------------------

I have tried to run Share Captioner but I have got CUDA Out of Memory error on a Rtx 3090:

`torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 172.00 MiB. GPU 0 has a total capacty of 24.00 GiB of which 0 bytes is free. Including non-PyTorch memory, this process has 17179869184.00 GiB memory in use. Of the allocated memory 23.10 GiB is allocated by PyTorch, and 3.24 MiB is reserved by PyTorch but unallocated. If reserved but unallocated memory is large try setting max_split_size_mb to avoid fragmentation. See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF`

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.