Hugging Face's logo

Spaces:
AlignmentResearch
/
tuned-lens
Running

App Files Community
48
levmckinney commited on
Commit
ffa4a7e
•
1 Parent(s): 218475e

pythia-rename (#27)

Browse files

- renamed pythia models to be consistent with new naming (3675bd5f442933d9d24ee973331ea7b8b69f0647)

Files changed (16) hide show
  1. __pycache__/app.cpython-310.pyc +0 -0
  2. app.py +3 -3
  3. lens/{pythia-1.4b-deduped → pythia-1.4b-deduped-v0}/config.json +0 -0
  4. lens/{pythia-1.4b-deduped → pythia-1.4b-deduped-v0}/params.pt +0 -0
  5. lens/{pythia-12b-deduped → pythia-12b-deduped-v0}/config.json +0 -0
  6. lens/{pythia-12b-deduped → pythia-12b-deduped-v0}/params.pt +0 -0
  7. lens/{pythia-160m-deduped → pythia-160m-deduped-v0}/config.json +0 -0
  8. lens/{pythia-160m-deduped → pythia-160m-deduped-v0}/params.pt +0 -0
  9. lens/{pythia-1b-deduped → pythia-1b-deduped-v0}/config.json +0 -0
  10. lens/{pythia-1b-deduped → pythia-1b-deduped-v0}/params.pt +0 -0
  11. lens/{pythia-410m-deduped → pythia-410m-deduped-v0}/config.json +0 -0
  12. lens/{pythia-410m-deduped → pythia-410m-deduped-v0}/params.pt +0 -0
  13. lens/{pythia-6.9b-deduped → pythia-6.9b-deduped-v0}/config.json +0 -0
  14. lens/{pythia-6.9b-deduped → pythia-6.9b-deduped-v0}/params.pt +0 -0
  15. lens/{pythia-70m-deduped → pythia-70m-deduped-v0}/config.json +0 -0
  16. lens/{pythia-70m-deduped → pythia-70m-deduped-v0}/params.pt +0 -0
__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