Sentdex
/

Walls1337bot-Llama2-7B-003.004.700

Text Generation

Model card Files Files and versions Community

Sentdex commited on Mar 1

Commit

191fbc5

•

1 Parent(s): f04ce00

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -14,7 +14,7 @@ from peft import AutoPeftModelForCausalLM
 from transformers import AutoTokenizer
 import torch
-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")

 from transformers import AutoTokenizer
 import torch
+model = AutoPeftModelForCausalLM.from_pretrained("Sentdex/Walls1337bot-Llama2-7B-003.004.700")
 tokenizer = AutoTokenizer.from_pretrained("NousResearch/Llama-2-7b-chat-hf")
 model = model.to("cuda")