pngwn HF Staff commited on
Commit
b158aae
·
verified ·
1 Parent(s): 8b11d32

Update ML Intern artifact metadata

Browse files
Files changed (1) hide show
  1. README.md +24 -3
README.md CHANGED
@@ -1,8 +1,9 @@
1
  ---
2
  tags:
3
- - sft
4
- - gsm8k
5
- - math
 
6
  base_model: Qwen/Qwen2.5-0.5B
7
  ---
8
 
@@ -33,3 +34,23 @@ Supervised fine-tuned model for grade-school math reasoning on GSM8K.
33
  ## Eval script
34
 
35
  The exact eval script used for both baseline and tuned evaluation is included in this repository as `eval_gsm8k.py`.
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  tags:
3
+ - sft
4
+ - gsm8k
5
+ - math
6
+ - ml-intern
7
  base_model: Qwen/Qwen2.5-0.5B
8
  ---
9
 
 
34
  ## Eval script
35
 
36
  The exact eval script used for both baseline and tuned evaluation is included in this repository as `eval_gsm8k.py`.
37
+
38
+ <!-- ml-intern-provenance -->
39
+ ## Generated by ML Intern
40
+
41
+ This model repository was generated by [ML Intern](https://github.com/huggingface/ml-intern), an agent for machine learning research and development on the Hugging Face Hub.
42
+
43
+ - Try ML Intern: https://smolagents-ml-intern.hf.space
44
+ - Source code: https://github.com/huggingface/ml-intern
45
+
46
+ ## Usage
47
+
48
+ ```python
49
+ from transformers import AutoModelForCausalLM, AutoTokenizer
50
+
51
+ model_id = 'pngwn/qwen2.5-0.5b-gsm8k-sft'
52
+ tokenizer = AutoTokenizer.from_pretrained(model_id)
53
+ model = AutoModelForCausalLM.from_pretrained(model_id)
54
+ ```
55
+
56
+ For non-causal architectures, replace `AutoModelForCausalLM` with the appropriate `AutoModel` class.