Spaces:
Running
Running
Commit
β’
ffa4a7e
1
Parent(s):
218475e
pythia-rename (#27)
Browse files- renamed pythia models to be consistent with new naming (3675bd5f442933d9d24ee973331ea7b8b69f0647)
- __pycache__/app.cpython-310.pyc +0 -0
- app.py +3 -3
- lens/{pythia-1.4b-deduped β pythia-1.4b-deduped-v0}/config.json +0 -0
- lens/{pythia-1.4b-deduped β pythia-1.4b-deduped-v0}/params.pt +0 -0
- lens/{pythia-12b-deduped β pythia-12b-deduped-v0}/config.json +0 -0
- lens/{pythia-12b-deduped β pythia-12b-deduped-v0}/params.pt +0 -0
- lens/{pythia-160m-deduped β pythia-160m-deduped-v0}/config.json +0 -0
- lens/{pythia-160m-deduped β pythia-160m-deduped-v0}/params.pt +0 -0
- lens/{pythia-1b-deduped β pythia-1b-deduped-v0}/config.json +0 -0
- lens/{pythia-1b-deduped β pythia-1b-deduped-v0}/params.pt +0 -0
- lens/{pythia-410m-deduped β pythia-410m-deduped-v0}/config.json +0 -0
- lens/{pythia-410m-deduped β pythia-410m-deduped-v0}/params.pt +0 -0
- lens/{pythia-6.9b-deduped β pythia-6.9b-deduped-v0}/config.json +0 -0
- lens/{pythia-6.9b-deduped β pythia-6.9b-deduped-v0}/params.pt +0 -0
- lens/{pythia-70m-deduped β pythia-70m-deduped-v0}/config.json +0 -0
- 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
|