a hacky fix for UCObjectStore?

mosaicml · Nov 8, 2023 · d97e4af · d97e4af
1 parent c1a80e6
commit d97e4af
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/llmfoundry/callbacks/hf_checkpointer.py b/llmfoundry/callbacks/hf_checkpointer.py
@@ -98,6 +98,7 @@ def __init__(
         if self.upload_to_object_store:
             self.remote_ud = RemoteUploaderDownloader(
                 bucket_uri=f'{self.backend}://{self.bucket_name}',
+                backend_kwargs={'path': self.save_dir_format_str},
                 num_concurrent_uploads=4)
         else:
             self.remote_ud = None