seonjeongh commited on
Commit
a976482
·
verified ·
1 Parent(s): f70e532

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -82,7 +82,7 @@ base_model = AutoModelForCausalLM.from_pretrained(lora_config.base_model_name_or
82
  device_map="auto")
83
  model = peft.PeftModel.from_pretrained(base_model, LORA_PATH)
84
  model = model.merge_and_unload(progressbar=True)
85
- tokenizer = AutoTokenizer.from_pretrained(LORA_PATH)
86
  ```
87
 
88
  ## Inference Example
 
82
  device_map="auto")
83
  model = peft.PeftModel.from_pretrained(base_model, LORA_PATH)
84
  model = model.merge_and_unload(progressbar=True)
85
+ tokenizer = AutoTokenizer.from_pretrained(lora_config.base_model_name_or_path)
86
  ```
87
 
88
  ## Inference Example