Visualization tool for various generation tasks on Language Models.
Project description
Conditional Language Model Generation Visualization
- when evaluating language models it is often pain to see what is generated and why
- this little package is a
vue.js
frontend together withflask
backend and it is designed to easily show some interesting visualizations on conditional generation models - it handles frontend-backend communication as well as frontend rendering
- hence the developper can focus only on ML aspects of his work!
VERSION: 0.2.1
changelog
- added
visuallm.elements.text_input_element.TextInputElement
- refactoring and better code quality
- resolved one dependabot alert
Installation
- install from pypi:
pip install visuallm
Usage
The documentation is a WIP as of now, however here you can see several snippets of what the library can do.
Generation Playground
Select which parameters you want to use for generation, plug in a HuggingFace
model, or an OpenAI
token and have fun with experimenting with various generation hyperparameters!
Chat Playground
Select which parameters you want to use for generation, plug in a HuggingFace
model, or an OpenAI
token and have fun with chatting with the model!
Visualize Next Token Predictions
By using visuallm.components.NextTokenPredictionComponent.NextTokenPredictionComponent
you can just plug the HuggingFace model in and go through the generation process step by step.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.