Unverified Commit 9bdce3a4 authored by Sam Shleifer's avatar Sam Shleifer Committed by GitHub
Browse files

[s2s] fix lockfile and peg distillation constants (#7545)

parent de4d7b00
......@@ -32,7 +32,7 @@ LAYERS_TO_COPY = {
},
16: { # maps num layers in student -> which teacher layers to copy
1: [0],
2: [0, 8],
2: [0, 15],
3: [0, 8, 15],
4: [0, 5, 10, 15],
6: [0, 3, 6, 9, 12, 15],
......
......@@ -11,7 +11,7 @@ except (ImportError, ModuleNotFoundError):
NLTK_AVAILABLE = False
if NLTK_AVAILABLE:
with FileLock("a_random_string") as lock:
with FileLock(".lock") as lock:
nltk.download("punkt", quiet=True)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment