Add library_name, pipeline_tag and link to paper/code

#1
by nielsr HF Staff - opened
Files changed (1) hide show
  1. README.md +24 -5
README.md CHANGED
@@ -1,18 +1,24 @@
1
  ---
2
- license: apache-2.0
 
3
  datasets:
4
  - AL-GR/AL-GR-v1
5
  language:
6
  - zh
 
7
  metrics:
8
  - accuracy
9
- base_model:
10
- - google-t5/t5-base
11
  ---
12
 
13
  # Forge-T5-Base-s1
14
 
15
- This model is initialized from the pre-trained [`google-t5/t5-base`](https://huggingface.co/google-t5/t5-base) and fine-tuned on the **AL-GR/AL-GR-v1** dataset using the [FORGE](https://github.com/AL-GR/FORGE) framework for **4 training epochs**.
 
 
 
 
16
 
17
  ## Evaluation Results on AL-GR/AL-GR-v1
18
 
@@ -58,4 +64,17 @@ After downloading, update the configuration file used by the FORGE framework. Sp
58
 
59
  ---
60
 
61
- For more details about the training setup, dataset, or evaluation protocol, please refer to the [FORGE framework repository](https://github.com/AL-GR/FORGE).
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
+ base_model:
3
+ - google-t5/t5-base
4
  datasets:
5
  - AL-GR/AL-GR-v1
6
  language:
7
  - zh
8
+ license: apache-2.0
9
  metrics:
10
  - accuracy
11
+ library_name: transformers
12
+ pipeline_tag: text-generation
13
  ---
14
 
15
  # Forge-T5-Base-s1
16
 
17
+ This model is initialized from the pre-trained [`google-t5/t5-base`](https://huggingface.co/google-t5/t5-base) and fine-tuned on the **AL-GR/AL-GR-v1** dataset using the [FORGE](https://github.com/selous123/al_sid) framework for **4 training epochs**.
18
+
19
+ More details can be found in the paper [FORGE: Forming Semantic Identifiers for Generative Retrieval in Industrial Datasets](https://huggingface.co/papers/2509.20904).
20
+
21
+ Official Code: [GitHub Repository](https://github.com/selous123/al_sid)
22
 
23
  ## Evaluation Results on AL-GR/AL-GR-v1
24
 
 
64
 
65
  ---
66
 
67
+ For more details about the training setup, dataset, or evaluation protocol, please refer to the [FORGE framework repository](https://github.com/selous123/al_sid).
68
+
69
+ ## Citation
70
+
71
+ If you find this work helpful, please cite the following paper:
72
+
73
+ ```bibtex
74
+ @article{fu2025forge,
75
+ title={FORGE: Forming Semantic Identifiers for Generative Retrieval in Industrial Datasets},
76
+ author={Fu, Kairui and Zhang, Tao and Xiao, Shuwen and Wang, Ziyang and Zhang, Xinming and Zhang, Chenchi and Yan, Yuliang and Zheng, Junjun and others},
77
+ journal={arXiv preprint arXiv:2509.20904},
78
+ year={2025}
79
+ }
80
+ ```