Commit Graph
26 Commits
Author SHA1 Message Date
49de587b2c Add optional support for dedicated endpoint (#4014)
Co-authored-by: Rayan Dasoriya <dasoriya@google.com>
2025-04-28 12:03:33 +00:00
9f4d837e54 Update peft code for stable_20250409 (#3991)
Co-authored-by: Rayan Dasoriya <dasoriya@google.com>
2025-04-15 12:29:40 +00:00
b58f02d5bf Add polling fn to common util (#3960)
Co-authored-by: Rayan Dasoriya <dasoriya@google.com>
2025-04-04 17:00:55 +00:00
040fc424fe Update common util (#3868)
Co-authored-by: Rayan Dasoriya <dasoriya@google.com>
2025-03-11 12:33:53 +00:00
b2d0a04114 Update source code for stable_20250213 peft docker image (#3864)
Co-authored-by: Rayan Dasoriya <dasoriya@google.com>
2025-03-06 13:56:56 +00:00
0943520f8d Update template location in dataset validation (#3799)
Co-authored-by: Rayan Dasoriya <dasoriya@google.com>
2025-01-23 13:26:44 +00:00
1f15188b6e Update common util and dataset validation util (#3796)
Co-authored-by: Rayan Dasoriya <dasoriya@google.com>
2025-01-23 02:27:23 +00:00
5d71616e8b Add ability to copy specific model artifacts (#3700)
Co-authored-by: Rayan Dasoriya <dasoriya@google.com>
2024-11-11 13:25:24 +00:00
57f598052e Update usage tracking metrics in the finetuning notebooks (#3695)
Co-authored-by: Rayan Dasoriya <dasoriya@google.com>
2024-11-11 13:24:27 +00:00
f11502ec80 Fix pip dependency error (#3683)
Co-authored-by: Rayan Dasoriya <dasoriya@google.com>
2024-10-30 12:38:10 +00:00
0727e19520 Add vmg templates, dataset_validation_util and update common_util (#3586)
* Add vmg templates, dataset_validation_util and update common_util

* Add name to CODEOWNERS

* Update common_util.py

---------

Co-authored-by: Rayan Dasoriya <dasoriya@google.com>
2024-09-26 13:16:09 +00:00
b98d0f7c29 Revert "Add chat completion to llama3.1 deployment notebook" (#3467)
This reverts commit 0a3ee2e642.

Co-authored-by: Rayan Dasoriya <dasoriya@google.com>
2024-08-22 14:11:47 +00:00
7af5359fa2 add codeowners for gemma2 finetuning (#3445)
Co-authored-by: Rayan Dasoriya <dasoriya@google.com>
2024-08-20 13:48:22 +00:00
c362269e75 feat: add support for mistral nemo (#3431)
Co-authored-by: Rayan Dasoriya <dasoriya@google.com>
2024-08-16 13:57:16 +00:00
cdca4a3c2e convert caption prompt to bool (#3422)
Co-authored-by: Rayan Dasoriya <dasoriya@google.com>
2024-08-13 21:13:56 +00:00
1ab9e4b6e3 Add common fn related to image task (#3417)
Co-authored-by: Rayan Dasoriya <dasoriya@google.com>
2024-08-12 14:33:12 +00:00
f5366faca7 Refactor paligemma and codegemma notebooks (#3408)
* refactor: refactor paligemma and codegemma notebooks

* fix lint issues

* fix lint issues

* fix lint issues

---------

Co-authored-by: Rayan Dasoriya <dasoriya@google.com>
2024-08-09 20:53:42 +00:00
cc53c8e9d9 refactor: refactor mistral and mixtral deployment notebooks (#3401)
Co-authored-by: Rayan Dasoriya <dasoriya@google.com>
2024-08-07 09:29:16 +00:00
7590168238 feat: add resize image function to common util (#3399)
Co-authored-by: Rayan Dasoriya <dasoriya@google.com>
2024-08-06 17:37:14 +00:00
c36e5f204d Update new job name creation format (#3377)
Co-authored-by: Rayan Dasoriya <dasoriya@google.com>
2024-08-02 20:55:38 +00:00
cc2011e354 Make mistral notebook tunable (#3374)
* Make mistral notebook tunable

* make minor improvements

* make minor improvements

---------

Co-authored-by: Rayan Dasoriya <dasoriya@google.com>
2024-08-01 13:47:50 +00:00
255b520c01 feat: add qwen2 deployment notebook (#3345)
* feat: add qwen2 deployment notebook

* fix lint issues

* Qwen2 additional minor improvements

* Fix lint issues

---------

Co-authored-by: Rayan Dasoriya <dasoriya@google.com>
2024-07-31 11:41:44 +00:00
671e9f84fa refactor: refactor gemma notebooks (#3255)
Co-authored-by: Rayan Dasoriya <dasoriya@google.com>
2024-07-11 19:23:15 +00:00
9387236c01 feat: add a fn to resize an image (#3246)
Co-authored-by: Rayan Dasoriya <dasoriya@google.com>
2024-07-11 17:30:47 +00:00
d57617c726 remove: remove model_garden_pytorch_mistral notebook (#3225)
Co-authored-by: Rayan Dasoriya <dasoriya@google.com>
2024-07-09 01:10:57 +00:00
cec3f9dd55 refactor: refactor llama3 deployment nb (#3100)
Co-authored-by: Rayan Dasoriya <dasoriya@google.com>
2024-06-18 12:36:46 +00:00