search

ndif-team / nnsight

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

Naming #89

Closed cadentj closed 6 months ago

cadentj commented 6 months ago 
from nnsight import LanguageModel

model = LanguageModel("openai-community/gpt2", unified=True, device="cuda:0")

print(model)