search

ndif-team / nnsight

The nnsight package enables interpreting and manipulating the internals of deep learned models.
https://nnsight.net/
MIT License
340 stars 34 forks source link

Make it easier to see what default inputs can go in `LanguageModel` #173

Closed loftusa closed 3 weeks ago

loftusa commented 1 month ago

model = LanguageModel("gpt2")

The help for LanguageModel doesn't say which other models are in the default namespace (for instance, can I do "llama3"? "gemma2"? etc). It would be nice to see a list somewhere.

This is especially helpful for knowing which models we can pass remote=True for

ericwtodd commented 1 month ago

To your second point, you can see which models are currently hosted on the nnsight website: https://nnsight.net/status/

JadenFiotto-Kaufman commented 1 month ago

@loftusa Do you mean remotely or just in general? The string like "openai-community/gpt2" is a huggingface repo id so whatever they support we support. You kinda just have to google what models are on huggingface.