From c1b741d9fb51c6e2d5bcda960524bbdd6fc21bf7 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Sun, 14 Jan 2024 17:31:51 -0500 Subject: [PATCH] pin model_revision for phi2 (#1123) --- examples/phi/phi2-ft.yml | 1 + 1 file changed, 1 insertion(+) diff --git a/examples/phi/phi2-ft.yml b/examples/phi/phi2-ft.yml index 24a1e5591f..81df941706 100644 --- a/examples/phi/phi2-ft.yml +++ b/examples/phi/phi2-ft.yml @@ -1,4 +1,5 @@ base_model: microsoft/phi-2 +model_revision: 834565c # pin model repo to the previous architecture model_type: AutoModelForCausalLM tokenizer_type: AutoTokenizer trust_remote_code: true