Skip to content

feat: LLM - Support for Batch Prediction for the textembedding models (preview)#2345

Merged
copybara-service[bot] merged 0 commit into
mainfrom
copybara_546112540
Jul 27, 2023
Merged

feat: LLM - Support for Batch Prediction for the textembedding models (preview)#2345
copybara-service[bot] merged 0 commit into
mainfrom
copybara_546112540

Conversation

@copybara-service
Copy link
Copy Markdown
Contributor

@copybara-service copybara-service Bot commented Jul 6, 2023 •
edited
Loading

feat: LLM - Support for Batch Prediction for the textembedding models (preview)

Usage:

model = TextEmbeddingModel.from_pretrained("textembedding-gecko@001")
job = model.batch_predict(
    dataset="gs://<bicket>/dataset.jsonl",
    destination_uri_prefix="gs://<bicket>/batch_prediction/",
    # Optional:
    model_parameters={},
)

@product-auto-label product-auto-label Bot added size: s Pull request size is small. api: vertex-ai Issues related to the googleapis/python-aiplatform API. labels Jul 6, 2023
@copybara-service copybara-service Bot force-pushed the copybara_546112540 branch from 02d5852 to 37bf2cf Compare July 27, 2023 00:50
@product-auto-label product-auto-label Bot added size: m Pull request size is medium. and removed size: s Pull request size is small. labels Jul 27, 2023
@copybara-service copybara-service Bot force-pushed the copybara_546112540 branch 2 times, most recently from a2f900d to b139e46 Compare July 27, 2023 21:15
@copybara-service copybara-service Bot closed this Jul 27, 2023
@copybara-service copybara-service Bot force-pushed the copybara_546112540 branch from b139e46 to a368538 Compare July 27, 2023 22:53
@copybara-service copybara-service Bot merged commit a368538 into main Jul 27, 2023
@product-auto-label product-auto-label Bot added size: u Pull request is empty. and removed size: m Pull request size is medium. labels Jul 27, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

No reviews

Assignees

No one assigned

Labels

api: vertex-ai Issues related to the googleapis/python-aiplatform API. google-contributor size: u Pull request is empty.

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

0 participants