Skip to content

Commit

Permalink
Merge pull request #260 from Butanium/patch-1
Browse files Browse the repository at this point in the history
Fix new synthax erro in readme
  • Loading branch information
JadenFiotto-Kaufman authored Oct 5, 2024
2 parents 6db86a0 + f12b043 commit 7d9bdaf
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -21,9 +21,9 @@ Here is a simple example where we run the nnsight API locally on gpt2 and save t
```python
from nnsight import LanguageModel

model = LanguageModel('openai-community/gpt2', device_map='auto') as tracer:
model = LanguageModel('openai-community/gpt2', device_map='auto')

with model.trace('The Eiffel Tower is in the city of'):
with model.trace('The Eiffel Tower is in the city of') as tracer:

hidden_states = model.transformer.h[-1].output[0].save()

Expand Down

1 comment on commit 7d9bdaf

@ReverseCache
Copy link

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Great work @Butanium !!

Please sign in to comment.