Lev McKinney commited on
Commit
3675bd5
β€’
1 Parent(s): 218475e

renamed pythia models to be consistent with new naming

Browse files
__pycache__/app.cpython-310.pyc ADDED
Binary file (3.92 kB). View file
 
app.py CHANGED
@@ -7,10 +7,10 @@ from plotly import graph_objects as go
7
 
8
  device = torch.device("cpu")
9
  print(f"Using device {device} for inference")
10
- model = AutoModelForCausalLM.from_pretrained("EleutherAI/pythia-410m-deduped")
11
  model = model.to(device)
12
- tokenizer = AutoTokenizer.from_pretrained("EleutherAI/pythia-410m-deduped")
13
- tuned_lens = TunedLens.load("pythia-410m-deduped", map_location=device)
14
  logit_lens = LogitLens(model)
15
 
16
  lens_options_dict = {
 
7
 
8
  device = torch.device("cpu")
9
  print(f"Using device {device} for inference")
10
+ model = AutoModelForCausalLM.from_pretrained("EleutherAI/pythia-410m-deduped-v0")
11
  model = model.to(device)
12
+ tokenizer = AutoTokenizer.from_pretrained("EleutherAI/pythia-410m-deduped-v0")
13
+ tuned_lens = TunedLens.load("pythia-410m-deduped-v0", map_location=device)
14
  logit_lens = LogitLens(model)
15
 
16
  lens_options_dict = {
lens/{pythia-1.4b-deduped β†’ pythia-1.4b-deduped-v0}/config.json RENAMED
File without changes
lens/{pythia-1.4b-deduped β†’ pythia-1.4b-deduped-v0}/params.pt RENAMED
File without changes
lens/{pythia-12b-deduped β†’ pythia-12b-deduped-v0}/config.json RENAMED
File without changes
lens/{pythia-12b-deduped β†’ pythia-12b-deduped-v0}/params.pt RENAMED
File without changes
lens/{pythia-160m-deduped β†’ pythia-160m-deduped-v0}/config.json RENAMED
File without changes
lens/{pythia-160m-deduped β†’ pythia-160m-deduped-v0}/params.pt RENAMED
File without changes
lens/{pythia-1b-deduped β†’ pythia-1b-deduped-v0}/config.json RENAMED
File without changes
lens/{pythia-1b-deduped β†’ pythia-1b-deduped-v0}/params.pt RENAMED
File without changes
lens/{pythia-410m-deduped β†’ pythia-410m-deduped-v0}/config.json RENAMED
File without changes
lens/{pythia-410m-deduped β†’ pythia-410m-deduped-v0}/params.pt RENAMED
File without changes
lens/{pythia-6.9b-deduped β†’ pythia-6.9b-deduped-v0}/config.json RENAMED
File without changes
lens/{pythia-6.9b-deduped β†’ pythia-6.9b-deduped-v0}/params.pt RENAMED
File without changes
lens/{pythia-70m-deduped β†’ pythia-70m-deduped-v0}/config.json RENAMED
File without changes
lens/{pythia-70m-deduped β†’ pythia-70m-deduped-v0}/params.pt RENAMED
File without changes