kelvin-t-lu commited on
Commit
1ed6ca7
1 Parent(s): c81b726

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -33,4 +33,4 @@ USER h2ogpt
33
 
34
  # ENTRYPOINT ["python3.10"]
35
 
36
- CMD ["python3.10", "generate.py", "--load_8bit", "True", "--base_model", "TheBloke/Llama-2-7B-Chat-GGML"]
 
33
 
34
  # ENTRYPOINT ["python3.10"]
35
 
36
+ CMD ["python3.10", "generate.py", "--load_8bit", "True", "--base_model", "h2oai/h2ogpt-4096-llama2-7b-chat"]