Skip to content

Commit ada574d

Browse files
HaiHui886haihwang
andauthored
Update deepseek yaml file (#43)
* push image to opencsg registry * add deep code model * update * update OpenCSG model name * update deepseek parameters --------- Co-authored-by: haihwang <[email protected]>
1 parent b886d5a commit ada574d

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

models/text-generation--opencsg--opencsg-deepseek-coder-1.3b-v0.1.yaml

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -32,9 +32,10 @@ model_config:
3232
do_sample: false
3333
top_k: 50
3434
top_p: 0.95
35+
repetition_penalty: 1
3536
num_return_sequences: 1
36-
max_new_tokens: 256
37-
prompt_format: "'role': 'user', 'content': {instruction}"
37+
max_new_tokens: 1024
38+
# prompt_format: "'role': 'user', 'content': {instruction}"
3839
scaling_config:
3940
num_workers: 1
4041
num_gpus_per_worker: 1

0 commit comments

Comments
 (0)