pin model_revision for phi2 (#1123)
Browse files- examples/phi/phi2-ft.yml +1 -0
examples/phi/phi2-ft.yml
CHANGED
|
@@ -1,4 +1,5 @@
|
|
| 1 |
base_model: microsoft/phi-2
|
|
|
|
| 2 |
model_type: AutoModelForCausalLM
|
| 3 |
tokenizer_type: AutoTokenizer
|
| 4 |
trust_remote_code: true
|
|
|
|
| 1 |
base_model: microsoft/phi-2
|
| 2 |
+
model_revision: 834565c # pin model repo to the previous architecture
|
| 3 |
model_type: AutoModelForCausalLM
|
| 4 |
tokenizer_type: AutoTokenizer
|
| 5 |
trust_remote_code: true
|