Views
No views yet
| Dataset | Task/Source | Size | Scenario |
|---|---|---|---|
| DISC-LawLLM-SFT-Pair | Legal information extraction | 32K | Legal professional assistant |
| Legal event detection | 27K | ||
| Legal case classification | 20K | ||
| Legal judgement prediction | 11K | ||
| Legal case matching | 8K | ||
| Legal text summarization | 9K | ||
| Judicial public opinion summarization | 6K | ||
| Legal question answering | 93K | Legal consultation services | |
| Legal reading comprehension | 38K | Judicial examination assistant | |
| Judicial examination | 12K | ||
| DISC-LawLLM-SFT-Triple | Legal judgement prediction | 16K | Legal professional assistant |
| Legal question answering | 23K | Legal consultation services | |
| General | Alpaca-GPT4 | 48K | General scenarios |
| Firefly | 60K | ||
| Total | 403K |
1>>>import torch
2>>>>>>from transformers import AutoModelForCausalLM, AutoTokenizer
3>>>from transformers.generation.utils import GenerationConfig
4>>>tokenizer = AutoTokenizer.from_pretrained("ShengbinYue/DISC-LawLLM", use_fast=False, trust_remote_code=True)
5>>>model = AutoModelForCausalLM.from_pretrained("ShengbinYue/DISC-LawLLM", device_map="auto", torch_dtype=torch.float16, trust_remote_code=True)
6>>>model.generation_config = GenerationConfig.from_pretrained("ShengbinYue/DISC-LawLLM")
7>>>messages = []
8>>>messages.append({"role": "user", "content": "生产销售假冒伪劣商品罪如何判刑?"})
9>>>response = model.chat(tokenizer, messages)
10>>>print(response)@misc{yue2023disclawllm,
title={DISC-LawLLM: Fine-tuning Large Language Models for Intelligent Legal Services},
author={Shengbin Yue and Wei Chen and Siyuan Wang and Bingxuan Li and Chenchen Shen and Shujun Liu and Yuxuan Zhou and Yao Xiao and Song Yun and Wei Lin and Xuanjing Huang and Zhongyu Wei},
year={2023},
eprint={2309.11325},
archivePrefix={arXiv},
primaryClass={cs.CL}
}
@inproceedings{yue2024lawllm,
title={LawLLM: Intelligent Legal System with Legal Reasoning and Verifiable Retrieval},
author={Yue, Shengbin and Liu, Shujun and Zhou, Yuxuan and Shen, Chenchen and Wang, Siyuan and Xiao, Yao and Li, Bingxuan and Song, Yun and Shen, Xiaoyu and Chen, Wei and others},
booktitle={International Conference on Database Systems for Advanced Applications},
pages={304--321},
year={2024},
organization={Springer}
}