Commit Graph

17 Commits

Author SHA1 Message Date
Googler 2e2ba9e5ea fix(components): Upload the tuned adapter to Model Registry instead of model checkpoint from `preview.llm.rlhf_pipeline`
PiperOrigin-RevId: 591393116
2023-12-15 16:47:51 -08:00
Googler c23b720f10 feat(components): Group `preview.llm.rlhf_pipeline` components for more readability
PiperOrigin-RevId: 582733817
2023-11-15 10:56:39 -08:00
Googler bcd59220f4 feat(components): Group `preview.llm.rlhf_pipeline` components for more readability
PiperOrigin-RevId: 582394326
2023-11-14 11:52:02 -08:00
Googler a927984394 feat(components): Group `preview.llm.rlhf_pipeline` components for more readability
PiperOrigin-RevId: 581354060
2023-11-10 13:32:31 -08:00
Googler f67cbfa81f feat(components): Add ability to tune chat model with `preview.llm.rlhf_pipeline`
PiperOrigin-RevId: 578262705
2023-10-31 12:15:30 -07:00
Googler d8f2c140ce feat(components): Add chat dataset preprocessor to `preview.llm.infer_pipeline`
PiperOrigin-RevId: 577988605
2023-10-30 16:20:21 -07:00
Googler e8fb6990df feat(components): Make `model_checkpoint` optional for `preview.llm.infer_pipeline`
PiperOrigin-RevId: 574876480
2023-10-19 08:14:58 -07:00
Googler 1386a826ba feat(components): Set display names for SFT, RLHF and LLM inference pipelines
PiperOrigin-RevId: 572897105
2023-10-12 07:17:39 -07:00
Googler 2bb57ece35 feat(components): Upload tensorboard metrics from RLHF pipeline if a tensorboard resource id is provided
PiperOrigin-RevId: 572040791
2023-10-09 14:26:17 -07:00
Googler 4fd1c02fc0 fix(components): Disable caching for LLM pipeline components that store temporary artifacts
PiperOrigin-RevId: 570070284
2023-10-02 08:14:00 -07:00
Connor McCarthy 1fbdeb2abf chore(components): fix GCPC markdown docstrings rendering
PiperOrigin-RevId: 565397565
2023-09-14 09:51:07 -07:00
Googler e21174f94a feat(components): Add sampling_strategy parameter to bulk inferrer to support different strategy. By default, we use greedy
PiperOrigin-RevId: 562860317
2023-09-05 12:43:17 -07:00
Connor McCarthy 3fb199658f chore(components): convert GCPC docstrings to markdown
PiperOrigin-RevId: 562015575
2023-09-01 12:12:27 -07:00
Googler 9ce2866527 feat(components): Update supported large model reference names that can be resolved by function based component in _implementation/llm
PiperOrigin-RevId: 559493244
2023-08-23 11:36:15 -07:00
Connor McCarthy 59aa41d78a chore(components): fix infer_pipeline invocation
PiperOrigin-RevId: 558586379
2023-08-20 10:03:56 -07:00
Connor McCarthy 7d282b91f4 chore(components): reorganize gcpc llm components
PiperOrigin-RevId: 558581292
2023-08-20 09:13:49 -07:00
Googler 611298a8ee feat(components): Add rlhf and infer pipelines to preview/llm. Add llm related components to _implementation/llm
PiperOrigin-RevId: 558430903
2023-08-19 11:38:14 -07:00