Hugging Face's logo

irlab-udc
/
LLaMA2-13b-Stop-Hate

Text Generation
PEFT
English
hate speech
conversational
Model card Files Community
palomapiot commited on
Commit
7647005
·
verified ·
1 Parent(s): 8d6792a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -35,7 +35,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, Conversation, pipe
35
 
36
  # Load the model
37
  config = PeftConfig.from_pretrained("irlab-udc/LLaMA2-13b-Stop-Hate")
38
- base_model = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-2-13b-chat-hf")
39
  model = PeftModel.from_pretrained(base_model, "irlab-udc/LLaMA2-13b-Stop-Hate")
40
  tokenizer = AutoTokenizer.from_pretrained("irlab-udc/LLaMA2-13b-Stop-Hate")
41
 
 
35
 
36
  # Load the model
37
  config = PeftConfig.from_pretrained("irlab-udc/LLaMA2-13b-Stop-Hate")
38
+ base_model = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-2-13b-chat-hf", config=config)
39
  model = PeftModel.from_pretrained(base_model, "irlab-udc/LLaMA2-13b-Stop-Hate")
40
  tokenizer = AutoTokenizer.from_pretrained("irlab-udc/LLaMA2-13b-Stop-Hate")
41