Felladrin commited on
Commit
ca11282
·
verified ·
1 Parent(s): 8355f3b

Upload README.md with huggingface_hub

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