Skip to content

Commit b4a5b84

Browse files
Fix stale VOLUME_NAMES[1] reference in model prep container
1 parent d530de1 commit b4a5b84

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

trainer/image_manager.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -520,7 +520,7 @@ def run_model_prep_container(
520520
command=command,
521521
labels=log_labels,
522522
environment=env,
523-
volumes={cst.VOLUME_NAMES[1]: {"bind": "/cache", "mode": "rw"}},
523+
volumes={cst.CACHE_VOLUME_NAME: {"bind": "/cache", "mode": "rw"}},
524524
device_requests=[docker.types.DeviceRequest(
525525
device_ids=[str(i) for i in gpu_ids],
526526
capabilities=[["gpu"]],

0 commit comments

Comments
 (0)