Konwertowanie modeli generatywnej AI PyTorch na potrzeby wnioskowania na urządzeniu

AI Edge Torch Generative API to wydajna biblioteka przeznaczona do tworzenia i konwertowania modeli PyTorch opartych na architekturze Transformer do formatu LiteRT/LiteRT-LM. Dzięki temu deweloperzy mogą łatwo wdrażać modele generatywnej AI, w szczególności duże modele językowe (LLM), do generowania tekstu i obrazów na urządzeniu.

Interfejs Torch Generative API obsługuje konwersję modeli na potrzeby wykonywania na procesorze i GPU, a obsługa NPU jest w trakcie opracowywania. Łącząc interfejs Torch Generative API z LiteRT-LM, możesz tworzyć responsywne aplikacje zapewniające ochronę prywatności, które uruchamiają modele generatywne w całości na urządzeniu.

Więcej informacji znajdziesz w repozytorium GitHub interfejsu Generative Torch API.