Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-02-24 12:10:09 +01:00
Code Issues Packages Projects Releases Wiki Activity

add debug info

Browse Source
This commit is contained in:
wangxingjun778
2025-09-10 11:00:06 +08:00
parent 6fe0443282
commit 3450db02e2
2 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
docker/build_image.py
View File

@@ -299,6 +299,7 @@ class LLMImageBuilder(Builder):
f'{self.args.torch_version} {self.args.torchvision_version} {self.args.torchaudio_version} '
f'{self.args.vllm_version} {self.args.lmdeploy_version} {self.args.autogptq_version} '
f'{self.args.flashattn_version}')
print(f'>>version_args: {version_args}')
with open('docker/Dockerfile.ubuntu', 'r') as f:
content = f.read()
content = content.replace('{base_image}', self.args.base_image)
@@ -319,7 +320,10 @@ class LLMImageBuilder(Builder):
self.args.modelscope_branch)
content = content.replace('{swift_branch}', self.args.swift_branch)
print('=' * 100)
print(f'>>Debug content for llm-image: \n{content}\n\n')
print('=' * 100)
print('\n\n')
return content
def image(self) -> str:

2
docker/install.sh
View File

@@ -30,7 +30,7 @@ if [[ "$(printf '%s\n' "0.6.0" "$vllm_version" | sort -V | head -n1)" = "0.6.0"
echo ">>Debug: Condition met: vllm_version >= 0.6.0, proceeding with installation..."
pip install --no-cache-dir vllm==$vllm_version && pip cache purge
else
echo ">>Debug: Condition not met: vllm_version < 0.6.0, skipping installation."
echo ">>Debug: Condition not met: vllm_version < 0.6.0, skipping installation. (vllm_version = $vllm_version)"
fi
# pip uninstall -y torch-scatter && TORCH_CUDA_ARCH_LIST="6.0;6.1;6.2;7.0;7.5;8.0;8.6;8.9;9.0" pip install --no-cache-dir -U torch-scatter