Update README.md

README.md

@@ -8,7 +8,7 @@ tags:
 pipeline_tag: text-generation
 ---
 
-# Berghof Vanilla 7B
+# Berghof NSFW 7B
 
 <img src="https://huggingface.co/Elizezen/Berghof-vanilla-7B/resolve/main/OIG1%20(2).jpg" alt="drawing" style="width:512px;"/>
 
@@ -24,9 +24,9 @@ Ensure you are using Transformers 4.34.0 or newer.
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
-tokenizer = AutoTokenizer.from_pretrained("Elizezen/Berghof-vanilla-7B")
+tokenizer = AutoTokenizer.from_pretrained("Elizezen/Berghof-NSFW-7B")
 model = AutoModelForCausalLM.from_pretrained(
-  "Elizezen/Berghof-vanilla-7B",
+  "Elizezen/Berghof-NSFW-7B",
   torch_dtype="auto",
 )
 model.eval()