Skip to content

Commit fd50714

Browse files
committed
Update Bert encoder and vocab paths
1 parent 4b5f3e0 commit fd50714

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

examples/text-generator/encoder.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -105,11 +105,9 @@ def decode(self, tokens):
105105

106106
def get_encoder():
107107
s3 = boto3.client("s3")
108-
encoder = json.load(
109-
s3.get_object(Bucket="cortex-examples", Key="gpt-2/124/encoder.json")["Body"]
110-
)
108+
encoder = json.load(s3.get_object(Bucket="cortex-examples", Key="gpt-2/encoder.json")["Body"])
111109
bpe_data = (
112-
s3.get_object(Bucket="cortex-examples", Key="gpt-2/124/vocab.bpe")["Body"]
110+
s3.get_object(Bucket="cortex-examples", Key="gpt-2/vocab.bpe")["Body"]
113111
.read()
114112
.decode("utf-8")
115113
)

0 commit comments

Comments
 (0)