Fix deprecation warning with Volume constructor

modal-labs · Mar 11, 2024 · 594b8c6 · 594b8c6
1 parent 6a18b37
commit 594b8c6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/common.py b/src/common.py
@@ -29,8 +29,8 @@
 stub = Stub(APP_NAME)
 
 # Volumes for pre-trained models and training runs.
-pretrained_volume = Volume.persisted("example-pretrained-vol")
-runs_volume = Volume.persisted("example-runs-vol")
+pretrained_volume = Volume.from_name("example-pretrained-vol", create_if_missing=True)
+runs_volume = Volume.from_name("example-runs-vol", create_if_missing=True)
 VOLUME_CONFIG: dict[str | PurePosixPath, Volume] = {
     "/pretrained": pretrained_volume,
     "/runs": runs_volume,