Skip to content

Commit 68428e4

Browse files
committed
Merge branch 'main' into bump-version
2 parents 9f3aee5 + 2b3a52f commit 68428e4

File tree

2 files changed

+3
-0
lines changed

2 files changed

+3
-0
lines changed

builder/manywheel/entrypoint_build.sh

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,8 @@ export PLAT_NAME=$PLAT_NAME
66
export USERID=${USERID}
77
export GROUPID=${GROUPID}
88
export CUDAVER=$(nvcc --version | sed -n 's/^.*release \([0-9]\+\).*$/\1/p')
9+
export NCCL_INCLUDE_DIR=/usr/local/cuda/include
10+
export NCCL_LIB_DIR=/usr/local/cuda/lib64
911

1012
source /opt/conda/bin/activate
1113
conda activate $PYTHON_VERSION

lmdeploy/pytorch/models/deepseek_vl2.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -109,6 +109,7 @@ def __init__(self,
109109
dtype: torch.dtype = None,
110110
device: torch.device = None):
111111
super().__init__()
112+
self.config = config
112113
self.ctx_mgr = ctx_mgr
113114

114115
# ----------- vision encoder ------------

0 commit comments

Comments
 (0)