Moving Phi-2 to HF provider (#89)

vriveras · web-flow · commit 5b4e9cd8318d · 2023-12-13T22:00:12.000-04:00
diff --git a/configs/phi-2/setup/project-settings.json b/configs/phi-2/setup/project-settings.json
@@ -204,7 +204,7 @@
         "label": "Enable gradient checkpointing:",
         "info": "Use gradient checkpointing. Recommended to save the memory.",
         "replaceToken": "<gradient_checkpointing>",
-        "defaultValue": true,
+        "defaultValue": false,
         "learnMore": "Can you tell me more about the Hugging Face trainer parameter gradient_checkpointing?"
       },
       {
diff --git a/index.json b/index.json
@@ -47,6 +47,11 @@
       "configPath": "olive-config.json",
       "workspaceTemplatePath": "configs/phi-2",
       "availabilityLocations": [
+        {
+          "providerType": "HuggingFace",
+          "modelId": "microsoft/phi-2",
+          "revision": "d3186761bf5c4409f7679359284066c25ab668ee"
+        },
         {
           "providerType": "Azure",
           "modelId": "azureml://registries/azureml-msr/models/microsoft-phi-2/versions/2",

Original file line number	Diff line number	Diff line change
`@@ -204,7 +204,7 @@`
`204`	`204`	`"label": "Enable gradient checkpointing:",`
`205`	`205`	`"info": "Use gradient checkpointing. Recommended to save the memory.",`
`206`	`206`	`"replaceToken": "<gradient_checkpointing>",`
`207`		`- "defaultValue": true,`
	`207`	`+ "defaultValue": false,`
`208`	`208`	`"learnMore": "Can you tell me more about the Hugging Face trainer parameter gradient_checkpointing?"`
`209`	`209`	`},`
`210`	`210`	`{`