fix function name

ai-dock · Jun 14, 2024 · 2e35bb1 · 2e35bb1
1 parent 594443e
commit 2e35bb1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/build/COPY_ROOT_0/opt/ai-dock/bin/build/layer0/cpu.sh b/build/COPY_ROOT_0/opt/ai-dock/bin/build/layer0/cpu.sh
@@ -5,7 +5,7 @@ build_cpu_main() {
     build_common_run_tests
 }
 
-build_cpu_install_deps() {
+build_cpu_install_torch() {
     "$PYTHON_VENV_PIP" install --no-cache-dir \
         torch==${PYTORCH_VERSION} \
         torchvision \