modelmesh-runtime-adapter icon indicating copy to clipboard operation
modelmesh-runtime-adapter copied to clipboard

[Feature Request] Reimplement Load Model of Triton and MLServer

Open WaterKnight1998 opened this issue 2 years ago • 0 comments

Good afternoon,

Thank you very much for creating this amazing framework.

I have seen a potential very good feature when doing inference with GPU models. I have seen that the implementation of triton and mlserver adapters use the following method: CalcMemCapacity to return model size.

This method returns model size based on disk size. However, for models executed in GPU it would be better to return the increase in VRAM. Do you think is doable? @tjohnson31415 @rafvasq @njhill @pvaneck

I am glad to help if you think is doable, but I don't have experience in GO, but I can learn

WaterKnight1998 avatar Jul 25 '23 14:07 WaterKnight1998