from transformers import pipeline

safe_pipe = pipeline(
    "text-generation",
    model="meta-llama/Llama-2-7b-chat-hf",
    torch_dtype="auto",
    device_map="auto"
)

naive_pipe = pipeline(
    "text-generation",
    model="microsoft/DialoGPT-medium",
    torch_dtype="auto",
    device_map="auto"
)

safe_out = safe_pipe(prompt, max_new_tokens=100, do_sample=False)[0]["generated_text"]
naive_out = naive_pipe(prompt, max_new_tokens=100, do_sample=False)[0]["generated_text"]

print("=== 安全对齐模型回答 ===")
print(safe_out)
print("\n=== 无对齐模型回答 ===")
print(naive_out)