Skip to content

Commit

Permalink
Added foundation model example notebooks
Browse files Browse the repository at this point in the history
  • Loading branch information
ryuta-yoshimatsu committed Jun 5, 2024
1 parent a12d9ef commit 83cfcbc
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions examples/foundation-model-examples/timesfm-example.py
Original file line number Diff line number Diff line change
Expand Up @@ -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",
],
Expand Down Expand Up @@ -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)

0 comments on commit 83cfcbc

Please sign in to comment.