Skip to content

Commit

Permalink
Some more noodling around to make click happy
Browse files Browse the repository at this point in the history
  • Loading branch information
dirkgr committed Apr 6, 2023
1 parent a3b969d commit 8cc0bb2
Showing 1 changed file with 5 additions and 0 deletions.
5 changes: 5 additions & 0 deletions scripts/upload_artifact.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,11 @@ def main(
wandb_run_path: str,
files_or_directories: Tuple[Path],
):
"""
Uploads artifacts to GCS. This uploads to a hardcoded bucket in GCS, because that's where we expect to keep all the artifacts for OLMo.
WANDB_RUN_PATH: The "Weights and Biases" run path. You get this by going to the run in wandb and clicking on the "copy run path" button. We will use this as the prefix for the paths in the GCS bucket.
"""
storage_client = storage.Client()
bucket = storage_client.bucket("allennlp-olmo", "ai2-allennlp")
prefix = wandb_run_path.strip("/")
Expand Down

0 comments on commit 8cc0bb2

Please sign in to comment.