switched to pythia-410m

app.py

@@ -7,10 +7,10 @@ from plotly import graph_objects as go
 
 device = torch.device("cpu")
 print(f"Using device {device} for inference")
-model = AutoModelForCausalLM.from_pretrained("gpt2")
+model = AutoModelForCausalLM.from_pretrained("EleutherAI/pythia-410m-deduped")
 model = model.to(device)
-tokenizer = AutoTokenizer.from_pretrained("gpt2")
-tuned_lens = TunedLens.load("lens/gpt2", map_location=device)
+tokenizer = AutoTokenizer.from_pretrained("EleutherAI/pythia-410m-deduped")
+tuned_lens = TunedLens.load("lens/pythia-410m-deduped", map_location=device)
 logit_lens = LogitLens(model)
 
 lens_options_dict = {