mirror of
https://github.com/fauxpilot/fauxpilot.git
synced 2025-07-05 20:41:55 -07:00
5 lines
279 B
Docker
5 lines
279 B
Docker
FROM moyix/triton_with_ft:22.09
|
|
|
|
# Install dependencies: torch
|
|
RUN python3 -m pip install --disable-pip-version-check -U torch --extra-index-url https://download.pytorch.org/whl/cu116
|
|
RUN python3 -m pip install --disable-pip-version-check -U transformers bitsandbytes accelerate
|