Browse Source

fix: update npu.Dockerfile to set TORCH_DEVICE_BACKEND_AUTOLOAD=0 for model download

myhloli 1 week ago
parent
commit
5f9fdd9b62
1 changed files with 1 additions and 1 deletions
  1. 1 1
      docker/china/npu.Dockerfile

+ 1 - 1
docker/china/npu.Dockerfile

@@ -22,7 +22,7 @@ RUN python3 -m pip install -U 'mineru[core]' -i https://mirrors.aliyun.com/pypi/
     python3 -m pip cache purge
 
 # Download models and update the configuration file
-RUN /bin/bash -c "mineru-models-download -s modelscope -m all"
+RUN TORCH_DEVICE_BACKEND_AUTOLOAD=0 /bin/bash -c "mineru-models-download -s modelscope -m all"
 
 # Set the entry point to activate the virtual environment and run the command line tool
 ENTRYPOINT ["/bin/bash", "-c", "export MINERU_MODEL_SOURCE=local && exec \"$@\"", "--"]