Load directly from *.pth?

[vigna]

From what I understand you code requires to convert first the *.pth files in a different format. Would it be possible to modify your code to load directly the *.pth files, as in the original Facebook LLaMa example? Sorry if the answer is trivial, I'm just finding my way.

[yrrah]

the native *.pth require a set number of GPUs per model, the conversion is a way around this limitation