Skip to content
GitLab
Commit f5fa449e authored by Мазур Грета Евгеньевна's avatar Мазур Грета Евгеньевна
Browse files

micro zapusk no cross

parent a5dae345
master
No related merge requests found
Hide whitespace changes
Inline Side-by-side
Showing
with 2 additions and 2 deletions
+2 -2
.ipynb_checkpoints/checkLora-checkpoint.py
+ 1
− 1
View file @ f5fa449e
......@@ -164,7 +164,7 @@ from transformers import AutoModel
config = PeftConfig.from_pretrained("micro_no_cross_fine_tuned")
model = AutoModel.from_pretrained(config.base_model_name_or_path)
model = PeftModel.from_pretrained(model, "micro_no_cross_fine_tuned")
model = PeftModel.from_pretrained(model, "micro_no_cross_fine_tuned", is_trainable = False)
model.to(device)
model.eval() # Переключение в режим предсказания
......
checkLora.py
+ 1
− 1
View file @ f5fa449e
......@@ -164,7 +164,7 @@ from transformers import AutoModel
config = PeftConfig.from_pretrained("micro_no_cross_fine_tuned")
model = AutoModel.from_pretrained(config.base_model_name_or_path)
model = PeftModel.from_pretrained(model, "micro_no_cross_fine_tuned")
model = PeftModel.from_pretrained(model, "micro_no_cross_fine_tuned", is_trainable = False)
model.to(device)
model.eval() # Переключение в режим предсказания
......
Supports Markdown
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