text / Dockerfile
MohitAron123's picture
Upload 2 files
1fd6075 verified
raw
history blame contribute delete
347 Bytes
# 拿一个干净的Python环境
FROM python:3.10
# 装好所有工具(加上bitsandbytes帮忙减肥,scipy帮忙算分)
RUN pip install --no-cache-dir fastapi uvicorn transformers torch accelerate bitsandbytes scipy
# 开门营业
EXPOSE 7860
# 用最顽强的Python启动方式,绝不中途罢工
CMD ["python", "main.py"]