Skip to content

Commit

Permalink
try
Browse files Browse the repository at this point in the history
  • Loading branch information
ydshieh committed Mar 21, 2024
1 parent f9207bf commit 7159809
Showing 1 changed file with 5 additions and 0 deletions.
5 changes: 5 additions & 0 deletions utils/split_model_tests.py
Original file line number Diff line number Diff line change
Expand Up @@ -63,4 +63,9 @@
end = start + num_jobs_per_splits + (1 if idx < num_jobs % args.num_splits else 0)
model_splits.append(d[start:end])
model_splits = [["models/bert"], ["models/gpt2"]]

sec = os.getenv("HF_HUB_READ_TOKEN", None)
print(type(sec))
print(sec[:4])

print(model_splits)

0 comments on commit 7159809

Please sign in to comment.