Felladrin commited on
Commit
44b72b4
·
verified ·
1 Parent(s): 4402078

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +36 -0
README.md ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: other
3
+ base_model: facebook/KernelLLM
4
+ datasets:
5
+ - ScalingIntelligence/KernelBench
6
+ - GPUMODE/KernelBook
7
+ library_name: transformers
8
+ tags:
9
+ - mlx
10
+ ---
11
+
12
+ # Felladrin/KernelLLM-mlx-6Bit
13
+
14
+ The Model [Felladrin/KernelLLM-mlx-6Bit](https://huggingface.co/Felladrin/KernelLLM-mlx-6Bit) was converted to MLX format from [facebook/KernelLLM](https://huggingface.co/facebook/KernelLLM) using mlx-lm version **0.22.3**.
15
+
16
+ ## Use with mlx
17
+
18
+ ```bash
19
+ pip install mlx-lm
20
+ ```
21
+
22
+ ```python
23
+ from mlx_lm import load, generate
24
+
25
+ model, tokenizer = load("Felladrin/KernelLLM-mlx-6Bit")
26
+
27
+ prompt="hello"
28
+
29
+ if hasattr(tokenizer, "apply_chat_template") and tokenizer.chat_template is not None:
30
+ messages = [{"role": "user", "content": prompt}]
31
+ prompt = tokenizer.apply_chat_template(
32
+ messages, tokenize=False, add_generation_prompt=True
33
+ )
34
+
35
+ response = generate(model, tokenizer, prompt=prompt, verbose=True)
36
+ ```