Views
No views yet
| Name | Quant method | Size |
|---|---|---|
| AIGCodeGeek-DS-6.7B.Q2_K.gguf | Q2_K | 2.36GB |
| AIGCodeGeek-DS-6.7B.Q3_K_S.gguf | Q3_K_S | 2.75GB |
| AIGCodeGeek-DS-6.7B.Q3_K.gguf | Q3_K | 3.07GB |
| AIGCodeGeek-DS-6.7B.Q3_K_M.gguf | Q3_K_M | 3.07GB |
| AIGCodeGeek-DS-6.7B.Q3_K_L.gguf | Q3_K_L | 3.35GB |
| AIGCodeGeek-DS-6.7B.IQ4_XS.gguf | IQ4_XS | 3.4GB |
| AIGCodeGeek-DS-6.7B.Q4_0.gguf | Q4_0 | 3.56GB |
| AIGCodeGeek-DS-6.7B.IQ4_NL.gguf | IQ4_NL | 3.59GB |
| AIGCodeGeek-DS-6.7B.Q4_K_S.gguf | Q4_K_S | 3.59GB |
| AIGCodeGeek-DS-6.7B.Q4_K.gguf | Q4_K | 3.8GB |
| AIGCodeGeek-DS-6.7B.Q4_K_M.gguf | Q4_K_M | 3.8GB |
| AIGCodeGeek-DS-6.7B.Q4_1.gguf | Q4_1 | 3.95GB |
| AIGCodeGeek-DS-6.7B.Q5_0.gguf | Q5_0 | 4.33GB |
| AIGCodeGeek-DS-6.7B.Q5_K_S.gguf | Q5_K_S | 4.33GB |
| AIGCodeGeek-DS-6.7B.Q5_K.gguf | Q5_K | 4.46GB |
| AIGCodeGeek-DS-6.7B.Q5_K_M.gguf | Q5_K_M | 4.46GB |
| AIGCodeGeek-DS-6.7B.Q5_1.gguf | Q5_1 | 4.72GB |
| AIGCodeGeek-DS-6.7B.Q6_K.gguf | Q6_K | 5.15GB |
| AIGCodeGeek-DS-6.7B.Q8_0.gguf | Q8_0 | 6.67GB |
1tokenizers>=0.14.0
2transformers>=4.35.0
3accelerate
4sympy>=1.12
5pebble
6timeout-decorator
7attrdict1import torch
2from transformers import AutoTokenizer, AutoModelForCausalLM
3tokenizer = AutoTokenizer.from_pretrained("aigcode/AIGCodeGeek-DS-6.7B", trust_remote_code=True)
4model = AutoModelForCausalLM.from_pretrained("aigcode/AIGCodeGeek-DS-6.7B", trust_remote_code=True, torch_dtype=torch.bfloat16).cuda()
5messages=[
6 { 'role': 'user', 'content': "write a merge sort algorithm in python."}
7]
8inputs = tokenizer.apply_chat_template(messages, add_generation_prompt=True, return_tensors="pt").to(model.device)
9# tokenizer.eos_token_id is the id of <|EOT|> token
10outputs = model.generate(inputs, max_new_tokens=512, do_sample=False, top_k=50, top_p=0.95, num_return_sequences=1, eos_token_id=tokenizer.eos_token_id)
11print(tokenizer.decode(outputs[0][len(inputs[0]):], skip_special_tokens=True))