mirror of
https://github.com/tatsu-lab/stanford_alpaca.git
synced 2024-10-01 05:35:37 -04:00
store tokenizer for recovered dir.
This commit is contained in:
parent
c53841268d
commit
7a95b21d2c
@ -133,6 +133,7 @@ def recover(
|
|||||||
|
|
||||||
if path_tuned is not None:
|
if path_tuned is not None:
|
||||||
model_recovered.save_pretrained(path_tuned)
|
model_recovered.save_pretrained(path_tuned)
|
||||||
|
tokenizer_recovered.save_pretrained(path_tuned)
|
||||||
|
|
||||||
if test_inference:
|
if test_inference:
|
||||||
input_text = (
|
input_text = (
|
||||||
|
Loading…
Reference in New Issue
Block a user