Update README.md

README.md

@@ -24,17 +24,17 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 
 # script config
 base_model_name = "LSX-UniWue/LLaMmlein_1B"
-chat_adapter_name = "LSX-UniWue/Betzerl_1B_wiki_preview"
+adapter_name = "LSX-UniWue/Betzerl_1B_wiki_preview"
 device = "cuda"  # or mps
 
 # load model
-config = PeftConfig.from_pretrained(chat_adapter_name)
+config = PeftConfig.from_pretrained(adapter_name)
 base_model = model = AutoModelForCausalLM.from_pretrained(
     base_model_name,
     torch_dtype=torch.bfloat16,
     device_map=device,
 )
 base_model.resize_token_embeddings(32001)
-model = PeftModel.from_pretrained(base_model, chat_adapter_name)
-tokenizer = AutoTokenizer.from_pretrained(chat_adapter_name)
+model = PeftModel.from_pretrained(base_model, adapter_name)
+tokenizer = AutoTokenizer.from_pretrained(adapter_name)
 ```