samhog
/

psychology-llama-rlhf

Model card Files Files and versions Community

samhog commited on Jun 19, 2023

Commit

f3bd8aa

•

1 Parent(s): ed64d59

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -27,7 +27,7 @@ model = LLaMAForCausalLM.from_pretrained(
     load_in_8bit=True,
     device_map="auto",
 )
-model = PeftModel.from_pretrained(model, "samhog/psychology-llama-rlaif")
 ```
 **Authors:**

     load_in_8bit=True,
     device_map="auto",
 )
+model = PeftModel.from_pretrained(model, "samhog/psychology-llama-rlhf")
 ```
 **Authors:**