docker: add options for CPU only, Intel GPU, AMD GPU (#5380)

This commit is contained in:
smCloudInTheSky 2024-01-28 15:18:14 +01:00 committed by GitHub
parent d921f80322
commit b1463df0a1
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
10 changed files with 252 additions and 46 deletions

View file

@ -20,3 +20,6 @@ APP_RUNTIME_GID=6972
# override default app build permissions (handy for deploying to cloud)
#APP_GID=6972
#APP_UID=6972
# Set cache env
TRANSFORMERS_CACHE=/home/app/text-generation-webui/cache/
HF_HOME=/home/app/text-generation-webui/cache/