Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model_name = "chenyiren/DeepSeek-7B-base_honeywords"
4tokenizer = AutoTokenizer.from_pretrained(model_name)
5model = AutoModelForCausalLM.from_pretrained(model_name)
1def evaluate_password(password):
2 prompt = f"Is the password '{password}' strong or weak? Explain why."
3 inputs = tokenizer(prompt, return_tensors="pt")
4 output = model.generate(**inputs, max_length=150)
5 print(tokenizer.decode(output[0], skip_special_tokens=True))
6
7evaluate_password("qwe123")1def generate_honeywords(password, num_honeywords=5):
2 prompt = f"Generate {num_honeywords} honeywords for the password '{password}'."
3 inputs = tokenizer(prompt, return_tensors="pt")
4 output = model.generate(**inputs, max_length=100)
5 print(tokenizer.decode(output[0], skip_special_tokens=True))
6
7generate_honeywords("P@ssw0rd!")huggingface/transformers https://github.com/JJJiangYH/LLaMA-Factory-