Vijayendra
/

QST-Llama-8b

Model card Files Files and versions Community

Vijayendra commited on Nov 11, 2024

Commit

342194b

·

verified ·

1 Parent(s): 1c86444

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -19,7 +19,7 @@ from peft import PeftModel, PeftConfig
 config = PeftConfig.from_pretrained("Vijayendra/QST-Llama-8b")
 base_model = AutoModelForCausalLM.from_pretrained("unsloth/llama-3-8b-bnb-4bit")
 model = PeftModel.from_pretrained(base_model, "Vijayendra/QST-Llama-8b")
-tokenizer = AutoTokenizer.from_pretrained("Vijayendra/llama-3b-merged-4bit")
 # Ensure padding token is set for the tokenizer
 if tokenizer.pad_token is None:

 config = PeftConfig.from_pretrained("Vijayendra/QST-Llama-8b")
 base_model = AutoModelForCausalLM.from_pretrained("unsloth/llama-3-8b-bnb-4bit")
 model = PeftModel.from_pretrained(base_model, "Vijayendra/QST-Llama-8b")
+tokenizer = AutoTokenizer.from_pretrained("Vijayendra/llama3.0-8B-merged-4bit")
 # Ensure padding token is set for the tokenizer
 if tokenizer.pad_token is None: