Spaces:
Paused
Paused
Commit
·
97f4954
1
Parent(s):
7732ddb
default model
Browse files- 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", "
|
|
|
33 |
|
34 |
# ENTRYPOINT ["python3.10"]
|
35 |
|
36 |
+
CMD ["python3.10", "generate.py", "--load_8bit", "True", "--base_model", "EleutherAI/gpt-j-6B"]
|