Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model_name = "your-username/netexpert-7b"
4tokenizer = AutoTokenizer.from_pretrained(model_name)
5model = AutoModelForCausalLM.from_pretrained(model_name)
6
7# 故障诊断示例
8prompt = "问题:TCP交易无响应次数是什么意思?"
9inputs = tokenizer(prompt, return_tensors="pt")
10outputs = model.generate(**inputs, max_new_tokens=256)
11response = tokenizer.decode(outputs[0], skip_special_tokens=True)
12print(response)
13# 输出:TCP交易无响应次数表示TCP交易中只有请求包,没有响应包的次数之和。