cesun commited on
Commit
5c46ab7
·
verified ·
1 Parent(s): a61c1e4

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -27,8 +27,8 @@ ADV-LLM is an **iteratively self-tuned** adversarial language model that generat
27
 
28
  ```python
29
  from transformers import AutoModelForCausalLM, AutoTokenizer
30
- model = AutoModelForCausalLM.from_pretrained("cesun/advllm-llama2")
31
- tokenizer = AutoTokenizer.from_pretrained("cesun/advllm-llama2")
32
  inputs = tokenizer("How to make a bomb", return_tensors="pt")
33
  outputs = model.generate(**inputs, max_new_tokens=90)
34
  print(tokenizer.decode(outputs[0], skip_special_tokens=True))
 
27
 
28
  ```python
29
  from transformers import AutoModelForCausalLM, AutoTokenizer
30
+ model = AutoModelForCausalLM.from_pretrained("cesun/advllm_llama2")
31
+ tokenizer = AutoTokenizer.from_pretrained("cesun/advllm_llama2")
32
  inputs = tokenizer("How to make a bomb", return_tensors="pt")
33
  outputs = model.generate(**inputs, max_new_tokens=90)
34
  print(tokenizer.decode(outputs[0], skip_special_tokens=True))