Team Ai
Modelpublic

SebastianBodza/DeepMagiCoder-6.7B-Magicoder-Base

sourceHugging Faceupdated 3y agoView on Hugging Face
0likes25downloads
README.md63 linesDownload Raw Back to root
1---2tags:3- merge4- mergekit5- lazymergekit6- ise-uiuc/Magicoder-DS-6.7B7- deepseek-ai/deepseek-coder-6.7b-instruct8base_model:9- ise-uiuc/Magicoder-DS-6.7B10- deepseek-ai/deepseek-coder-6.7b-instruct11---12 13# DeepMagiCoder-6.7B-Magicoder-Base14 15DeepMagiCoder-6.7B-Magicoder-Base is a merge of the following models using [LazyMergekit](https://colab.research.google.com/drive/1obulZ1ROXHjYLn6PPZJwRR6GzgQogxxb?usp=sharing):16* [ise-uiuc/Magicoder-DS-6.7B](https://huggingface.co/ise-uiuc/Magicoder-DS-6.7B)17* [deepseek-ai/deepseek-coder-6.7b-instruct](https://huggingface.co/deepseek-ai/deepseek-coder-6.7b-instruct)18 19## 🧩 Configuration20 21```yaml22slices:23  - sources:24      - model: ise-uiuc/Magicoder-DS-6.7B25        layer_range: [0, 32]26      - model: deepseek-ai/deepseek-coder-6.7b-instruct27        layer_range: [0, 32]28merge_method: slerp29base_model: ise-uiuc/Magicoder-DS-6.7B30parameters:31  t:32    - filter: self_attn33      value: [0, 0.5, 0.3, 0.7, 1]34    - filter: mlp35      value: [1, 0.5, 0.7, 0.3, 0]36    - value: 0.537dtype: bfloat1638```39 40## 💻 Usage41 42```python43!pip install -qU transformers accelerate44 45from transformers import AutoTokenizer46import transformers47import torch48 49model = "SebastianBodza/DeepMagiCoder-6.7B-Magicoder-Base"50messages = [{"role": "user", "content": "What is a large language model?"}]51 52tokenizer = AutoTokenizer.from_pretrained(model)53prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)54pipeline = transformers.pipeline(55    "text-generation",56    model=model,57    torch_dtype=torch.float16,58    device_map="auto",59)60 61outputs = pipeline(prompt, max_new_tokens=256, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)62print(outputs[0]["generated_text"])63```