is it possible to use it with other models?

poloclub / transformer-explainer

Transformer Explained Visually: Learn How LLM Transformer Models Work with Interactive Visualization

https://poloclub.github.io/transformer-explainer/

MIT License

3.11k stars 278 forks source link

is it possible to use it with other models? #13

Closed deepakdeore2004 closed 2 months ago

deepakdeore2004 commented 2 months ago

looks like the the visualization is with gpt2 model which is included in repo itself is it possible to run this app with custom model endpoint?

LearningForeverMS commented 2 months ago

I also want to know if we can integrate with other llm model. What all code modifications might be required ?

gracekimcy commented 2 months ago

Thanks for your interest in Transformer Explainer’s potential to support more models! The ManimML project from our group has been investigating how to enable ML developers more easily create animated visual explanations of ML concepts https://github.com/helblazer811/ManimML . Currently Transformer Explainer does not yet support similar capabilities, but that’s something on our long-term road map.