irlab-udc
/

Mistral-7b-Stop-Hate

Text Generation

Model card Files Files and versions Community

palomapiot commited on Jun 19

Commit

a6bede0

•

1 Parent(s): 2c6e48b

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -35,7 +35,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, Conversation, pipe
 # Load the model
 config = PeftConfig.from_pretrained("irlab-udc/Mistral-7b-Stop-Hate")
-base_model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-Instruct-v0.2")
 model = PeftModel.from_pretrained(base_model, "irlab-udc/Mistral-7b-Stop-Hate")
 tokenizer = AutoTokenizer.from_pretrained("irlab-udc/Mistral-7b-Stop-Hate")

 # Load the model
 config = PeftConfig.from_pretrained("irlab-udc/Mistral-7b-Stop-Hate")
+base_model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-Instruct-v0.2", config=config)
 model = PeftModel.from_pretrained(base_model, "irlab-udc/Mistral-7b-Stop-Hate")
 tokenizer = AutoTokenizer.from_pretrained("irlab-udc/Mistral-7b-Stop-Hate")