update swift docker

This commit is contained in:
Jintao Huang
2025-09-16 00:07:44 +08:00
parent 0bcefbe08a
commit 4ebfecb6cb
4 changed files with 7 additions and 6 deletions

View File

@@ -79,8 +79,8 @@ RUN if [ "$INSTALL_MS_DEPS" = "True" ]; then \
pip install --no-cache-dir huggingface-hub transformers peft -U; \
fi; \
if [ "$INSTALL_MEGATRON_DEPS" = "True" ]; then \
pip install "sglang[all]<0.4.10" "math_verify==0.5.2" "gradio<5.33" -U && \
pip install liger_kernel nvitop pre-commit "transformers<4.55" huggingface-hub -U && \
pip install "sglang[all]<0.5" "math_verify==0.5.2" "gradio<5.33" -U && \
pip install liger_kernel wandb swanlab nvitop pre-commit "transformers<4.57" "trl<0.21" huggingface-hub -U && \
SITE_PACKAGES=$(python -c "import site; print(site.getsitepackages()[0])") && echo $SITE_PACKAGES && \
CUDNN_PATH=$SITE_PACKAGES/nvidia/cudnn CPLUS_INCLUDE_PATH=$SITE_PACKAGES/nvidia/cudnn/include \
pip install --no-build-isolation transformer_engine[pytorch]; \

View File

@@ -354,9 +354,9 @@ class SwiftImageBuilder(LLMImageBuilder):
args.torchaudio_version = '2.7.1'
args.torchvision_version = '0.22.1'
if not args.vllm_version:
args.vllm_version = '0.10.0'
args.vllm_version = '0.10.1.1'
if not args.lmdeploy_version:
args.lmdeploy_version = '0.9.2'
args.lmdeploy_version = '0.9.2.post1'
if not args.flashattn_version:
args.flashattn_version = '2.7.4.post1'
return super().init_args(args)

View File

@@ -5,11 +5,11 @@ import hashlib
import os
import sys
import time
import zoneinfo
from datetime import datetime
from pathlib import Path
from typing import Generator, List, Optional, Union
import zoneinfo
from filelock import BaseFileLock, FileLock, SoftFileLock, Timeout
from modelscope.hub.constants import (DEFAULT_MODELSCOPE_DOMAIN,

View File

@@ -3,9 +3,10 @@
import datetime
import unittest
import zoneinfo
from unittest.mock import Mock, patch
import zoneinfo
from modelscope.hub.api import DatasetInfo, HubApi, ModelInfo
from modelscope.utils.constant import REPO_TYPE_DATASET, REPO_TYPE_MODEL
from modelscope.utils.repo_utils import DetailedCommitInfo