From f12b0435d8dba19f040324ac1dde77a122910982 Mon Sep 17 00:00:00 2001 From: Clement Dumas Date: Sat, 5 Oct 2024 01:48:58 +0200 Subject: [PATCH] Fix new synthax erro in readme --- README.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 671da635..bab96b82 100644 --- a/README.md +++ b/README.md @@ -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()