mirror of
https://github.com/modelscope/modelscope.git
synced 2026-02-24 04:01:10 +01:00
update swift docker
This commit is contained in:
@@ -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]; \
|
||||
|
||||
@@ -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)
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user