法律大模型：面向法律领域的专用语言模型

python from transformers import AutoModelForCausalLM, AutoTokenizer model_name = "your-legal-model" tokenizer = AutoTokenizer.from_pretrained(model_name) model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto") prompt = "根据以下事实，判断是否构成工伤：张三在上班途中被非本人主要责任的机动车撞伤。" inputs = tokenizer(f"<法律咨询> {prompt} </法律咨询>", return_tensors="pt").to("cuda") outputs = model.generate(**inputs, max_length=512) print(tokenizer.decode(outputs[0]))