"...git@developer.sourcefind.cn:chenpangpang/open-webui.git" did not exist on "e5dd7e65d48dfa3f6eaf158a59388c0e99cf3f47"
Commit a7b4cfe9 authored by carefree0910's avatar carefree0910 Committed by Lysandre Debut
Browse files

Update README.md

I assume that it should test the `re-load` functionality after testing the `save` functionality, however I'm also surprised that nobody points this out after such a long time, so maybe I've misunderstood the purpose. This PR is just in case :)
parent aaedfc35
...@@ -123,7 +123,7 @@ traced_model = torch.jit.trace(model, (input_ids,)) ...@@ -123,7 +123,7 @@ traced_model = torch.jit.trace(model, (input_ids,))
model.save_pretrained('./directory/to/save/') # save model.save_pretrained('./directory/to/save/') # save
model = model_class.from_pretrained('./directory/to/save/') # re-load model = model_class.from_pretrained('./directory/to/save/') # re-load
tokenizer.save_pretrained('./directory/to/save/') # save tokenizer.save_pretrained('./directory/to/save/') # save
tokenizer = tokenizer_class.from_pretrained(pretrained_weights) tokenizer = tokenizer_class.from_pretrained('./directory/to/save/') # re-load
# SOTA examples for GLUE, SQUAD, text generation... # SOTA examples for GLUE, SQUAD, text generation...
``` ```
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment