Merge pull request #52 from databricks-industry-solutions/foundation-…

…model-notebooks Added foundation model example notebooks
databricks-industry-solutions · Jun 5, 2024 · 3e5ae1c · 3e5ae1c
2 parents 0cabb62 + 83cfcbc
commit 3e5ae1c
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/examples/foundation-model-examples/timesfm-example.py b/examples/foundation-model-examples/timesfm-example.py
@@ -159,7 +159,6 @@ def __setstate__(self, state):
     input_example=df,
     pip_requirements=[
       "jax[cuda12]==0.4.26",
-      "protobuf==3.20.*",
       "utilsforecast==0.1.10",
       "git+https://github.com/google-research/timesfm.git",
     ],
@@ -375,4 +374,4 @@ def forecast(input_data, url=endpoint_url, databricks_token=token):
 
 # COMMAND ----------
 
-func_delete_model_serving_endpoint(model_serving_endpoint_name)
+#func_delete_model_serving_endpoint(model_serving_endpoint_name)