Update README.md

README.md

@@ -1,7 +1,11 @@
 ---
 license: apache-2.0
+datasets:
+- Sentdex/WSB-003.004
 ---
 
+Probably don't use this model, I'm just tinkering, but it's a multi-turn, multi-speaker model attempt trained from /r/wallstreetbets data that you can find: https://huggingface.co/datasets/Sentdex/WSB-003.004
+
 ```py
 #https://huggingface.co/docs/peft/quicktour 
 
@@ -9,7 +13,7 @@ from peft import AutoPeftModelForCausalLM
 from transformers import AutoTokenizer
 import torch
 
-model = AutoPeftModelForCausalLM.from_pretrained("Sentdex/Walls1337bot-Llama2-7B-003.004.400")
+model = AutoPeftModelForCausalLM.from_pretrained("Sentdex/Walls1337bot-Llama2-7B-003.004.500")
 tokenizer = AutoTokenizer.from_pretrained("NousResearch/Llama-2-7b-chat-hf")
 
 model = model.to("cuda")