Skip to content
This repository has been archived by the owner on Jun 22, 2024. It is now read-only.

Commit

Permalink
Merge pull request #13 from luoweb/main
Browse files Browse the repository at this point in the history
Add baichuan 7b and wizardcode
  • Loading branch information
mudler authored Jul 25, 2023
2 parents 51b8acf + c1dec85 commit fd2f787
Show file tree
Hide file tree
Showing 2 changed files with 79 additions and 0 deletions.
34 changes: 34 additions & 0 deletions baichuan-7b.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,34 @@
name: "baichuan-7b"

description: |
Baichuan-7B is an open-source large-scale pre-trained model developed by Baichuan Intelligent Technology. Based on the Transformer architecture, it is a model with 7 billion parameters trained on approximately 1.2 trillion tokens. It supports both Chinese and English, with a context window length of 4096.
license: "Apache 2.0"
urls:
- https://github.com/baichuan-inc/Baichuan-7B

config_file: |
backend: llama
parameters:
model: baichuan-vicuna-7b.ggmlv3.q4_0.bin
top_k: 80
temperature: 0.2
top_p: 0.7
context_size: 1024
template:
completion: baichuan-completion
chat: baichuan-chat
files:
- filename: "baichuan-vicuna-7b.ggmlv3.q4_0.bin"
sha256: "d7c12bccc40098830a18d2f3a42020d243396d240b6020a0f2daace6b203d444"
uri: "https://huggingface.co/TheBloke/baichuan-vicuna-7B-GGML/resolve/main/baichuan-vicuna-7b.ggmlv3.q4_0.bin"

prompt_templates:
- name: "baichuan-completion"
content: |
USER: Complete the following text: {{.Input}}
ASSISTANT:
- name: "baichuan-chat"
content: |
USER: {{.Input}}
ASSISTANT:
45 changes: 45 additions & 0 deletions wizardcode-15b.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,45 @@
name: "wizardcode-15b"

description: |
Empowering Code Large Language Models with Evol-Instruct
license: "Apache 2.0"

urls:
- https://github.com/nlpxucan/WizardLM

config_file: |
name: wizardcode13b
# Default model parameters
parameters:
# Relative to the models path
model: WizardCoder-15B-1.0.ggmlv3.q4_1.bin
temperature: 0.9
top_k: 50
top_p: 0.95
backend: starcoder
context_size: 8192
template:
completion: wizardcode-completion
chat: wizardcode-chat
stopwords:
- "<unk>"
- <|endoftext|>
- <|end|>
files:
- filename: "WizardCoder-15B-1.0.ggmlv3.q4_0.bin"
sha256: "b70164bc0b58a472c0987905133735ab3b27e2c439dedf8174a43951c51c3229"
uri: "https://huggingface.co/TheBloke/WizardCoder-Guanaco-15B-V1.0-GGML/resolve/main/WizardCoder-15B-1.0.ggmlv3.q4_0.bin"

prompt_templates:
- name: "wizardcode-completion"
content: |
{{.Input}}
- name: "wizardcode-chat"
content: |
Below is an instruction that describes a task. Write a response that appropriately completes the request
### Instruction: {{.Input}}
### Response:

0 comments on commit fd2f787

Please sign in to comment.