vmt icon indicating copy to clipboard operation
vmt copied to clipboard

VRAM usage

Open jiyuwangbupt opened this issue 1 year ago • 0 comments

May I ask how much VRAM is needed at a minimum to perform inference and training for this model? I am using an A6000 with 46GB of VRAM, and I encountered an out-of-memory error while running python test.py during the installation process.

jiyuwangbupt avatar Nov 13 '24 12:11 jiyuwangbupt