diff --git a/.ipynb_checkpoints/checkLora-checkpoint.py b/.ipynb_checkpoints/checkLora-checkpoint.py index e230dbae6110494e98bdf2efcf4b21d0654f5b9d..f6467ee1ee31aca7b6f709a235c399411e6b1435 100644 --- a/.ipynb_checkpoints/checkLora-checkpoint.py +++ b/.ipynb_checkpoints/checkLora-checkpoint.py @@ -299,7 +299,7 @@ BASE_MODEL_PATH = "./micro_no_cross_fine_tuned/base" LORA_PATH = "./micro_no_cross_fine_tuned/lora" # Загружаем токенизатор -tokenizer = BertTokenizer.from_pretrained(BASE_MODEL_PATH) +tokenizer = BertTokenizer.from_pretrained("./micro_no_cross_fine_tuned") # Загружаем базовую модель base_model = BertForSequenceClassification.from_pretrained(BASE_MODEL_PATH) diff --git a/checkLora.py b/checkLora.py index e230dbae6110494e98bdf2efcf4b21d0654f5b9d..f6467ee1ee31aca7b6f709a235c399411e6b1435 100644 --- a/checkLora.py +++ b/checkLora.py @@ -299,7 +299,7 @@ BASE_MODEL_PATH = "./micro_no_cross_fine_tuned/base" LORA_PATH = "./micro_no_cross_fine_tuned/lora" # Загружаем токенизатор -tokenizer = BertTokenizer.from_pretrained(BASE_MODEL_PATH) +tokenizer = BertTokenizer.from_pretrained("./micro_no_cross_fine_tuned") # Загружаем базовую модель base_model = BertForSequenceClassification.from_pretrained(BASE_MODEL_PATH)