guchuntao

> I have loaded the model into the gpu and I can see with nvidia-smi that the memory is increased, but it is still running on the cpu