Training GPU

[lingxitong]

Hello, Nice work, can you tell the training gpu information?

[ccitachi]

Hello, thanks for your interest! The model was trained on a single NVIDIA RTX A6000 GPU with 48 GB of VRAM.