Add experimental support for CUDA (#38)
Signed-off-by:
Hung-Han (Henry) Chen <chenhungh@gmail.com>
Showing
- .github/workflows/cuda_image.yaml 76 additions, 0 deletions.github/workflows/cuda_image.yaml
- Dockerfile.cuda11 13 additions, 0 deletionsDockerfile.cuda11
- Dockerfile.cuda12 12 additions, 0 deletionsDockerfile.cuda12
- README.md 26 additions, 4 deletionsREADME.md
- charts/ialacol/Chart.yaml 2 additions, 2 deletionscharts/ialacol/Chart.yaml
- charts/ialacol/values.yaml 1 addition, 0 deletionscharts/ialacol/values.yaml
- examples/values/llama2-7b-chat-cuda11.yaml 30 additions, 0 deletionsexamples/values/llama2-7b-chat-cuda11.yaml
- get_config.py 4 additions, 0 deletionsget_config.py
- get_llm.py 7 additions, 5 deletionsget_llm.py
- model_generate.py 4 additions, 0 deletionsmodel_generate.py
- requirements.txt 1 addition, 1 deletionrequirements.txt
Loading
Please register or sign in to comment