search

niuzaisheng / ScreenAgent

ScreenAgent: A Computer Control Agent Driven by Visual Language Large Model (IJCAI-24)
https://arxiv.org/abs/2402.07945
Other
273 stars 27 forks source link

GPU Memroy #23

Open irfanfadhullah opened 4 months ago

irfanfadhullah commented 4 months ago

Hello, I just wondering how manu GPU that we need to run this model? And is it support quantization method?

Mikael17125 commented 4 months ago

yes, it support quantization, you can modify this part https://github.com/niuzaisheng/ScreenAgent/blob/2b490c5d41a4a3301daad1ea135748b2b9d72c4c/train/cogagent_model_worker.py#L51

PhamQuocHuy1101 commented 3 weeks ago

Hello, and to re-produce model, how many GPU need for training?