Instructions to use diffusionfamily/diffugpt-m with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use diffusionfamily/diffugpt-m with Transformers:
# Load model directly from transformers import AutoTokenizer, AutoModel tokenizer = AutoTokenizer.from_pretrained("diffusionfamily/diffugpt-m") model = AutoModel.from_pretrained("diffusionfamily/diffugpt-m") - Notebooks
- Google Colab
- Kaggle
metadata
library_name: transformers
license: apache-2.0
base_model: gpt2-medium
tags:
- llama-factory
- full
- diffusion
model-index:
- name: diffugpt-m
results: []
datasets:
- HuggingFaceFW/fineweb
language:
- en
diffugpt-m
This model is a fine-tuned version of gpt2-medium on an unknown dataset.
Model description
Details and model loading can be seen https://github.com/HKUNLP/DiffuLLaMA.
Framework versions
- Transformers 4.44.2
- Pytorch 2.1.1+cu121
- Datasets 2.21.0
- Tokenizers 0.19.1
@misc{gong2024scalingdiffusionlanguagemodels,
title={Scaling Diffusion Language Models via Adaptation from Autoregressive Models},
author={Shansan Gong and Shivam Agarwal and Yizhe Zhang and Jiacheng Ye and Lin Zheng and Mukai Li and Chenxin An and Peilin Zhao and Wei Bi and Jiawei Han and Hao Peng and Lingpeng Kong},
year={2024},
eprint={2410.17891},
archivePrefix={arXiv},
primaryClass={cs.CL},
url={https://arxiv.org/abs/2410.17891},
}