1 repository
Maintains independent parser states for multiple concurrent language model generation streams within a single batch.
Distinct from Batch Request Execution: Distinct from API request batching: specifically manages state-machine parser contexts for concurrent LLM token generation.
Explore 1 awesome GitHub repository matching web development · Generation Stream Batching. Refine with filters or upvote what's useful.
Această bibliotecă oferă un framework pentru impunerea constrângerilor structurale asupra output-ului modelelor de limbaj în timpul procesului de generare a token-urilor. Funcționează ca un middleware care restricționează răspunsurile modelului pentru a adera strict la scheme JSON predefinite sau la tipare de expresii regulate, asigurându-se că textul generat este lizibil pentru mașină și consistent pentru consumul ulterior de date. Proiectul se distinge prin integrarea directă cu motoarele de inferență pentru a intercepta distribuțiile de probabilitate ale token-urilor înainte de etapa finală de eșantionare. Prin utilizarea parsării cu mașini de stare și descompunerea recursivă a schemelor, efectuează validarea lookahead pentru a elimina secvențele de token-uri invalide. Această abordare permite un control precis asupra output-ului, inclusiv impunerea unei ordini specifice a câmpurilor în obiectele JSON și capacitatea de a gestiona mai multe fluxuri de generare concurente prin execuția constrângerilor în loturi (batched). Biblioteca suportă o gamă largă de strategii de integrare, permițându-i să funcționeze pe diverse backend-uri de modele și medii de server de inferență. Include instrumente de diagnosticare pentru analizarea impactului acestor constrângeri asupra performanței, asigurând compatibilitatea și eficiența pe diferite configurații hardware. Software-ul este distribuit ca un pachet Python conceput pentru integrarea în pipeline-urile de inferență existente.
Enables efficient handling of multiple concurrent generation streams by maintaining separate parser states for each request.