default install tf-keras (#1064)

This commit is contained in:
tastelikefeet
2024-11-04 20:15:18 +08:00
committed by GitHub
parent 5e78f92e24
commit 43536064d5

View File

@@ -48,6 +48,8 @@ RUN pip install --no-cache-dir torch=={torch_version} torchvision=={torchvision_
RUN pip install --no-cache-dir transformers -U huggingface-hub==0.25.0
RUN pip install tf-keras==2.16.0 --no-dependencies
RUN pip config set global.index-url https://mirrors.aliyun.com/pypi/simple && \
pip config set install.trusted-host mirrors.aliyun.com && \
cp /tmp/resources/ubuntu2204.aliyun /etc/apt/sources.list