apache / apache/beam

[Bug]: Model loading repeatedly fails for large models in RunInference

Open
#25,286 0 comments 0 reactions 0 assignees View on GitHub
bug dataflow ml P3 python run-inference
Dominant language
Java
Stars
8.7k
Forks
4.7k
Avg merge
1d 20h
Merged PRs (30d)
196

Description

### What happened?

Right now in RunInference, when loading large models from remote locations (e.g. gcs), we timeout our request and eventually kill the work item/try a new one. We should have some mechanism for loading large remote models without timing out.

Note that the recommended path for large models will mostly be building a custom container, so this isn't a _huge_ deal, but that doesn't play well with model updates or pulling from model registries.

You can reproduce this by trying to load the t5-11b model remotely instead of from a custom container (https://beam.apache.org/documentation/ml/large-language-modeling/)

### Issue Priority

Priority: 3 (minor)

### Issue Components

- [X] Component: Python SDK
- [ ] Component: Java SDK
- [ ] Component: Go SDK
- [ ] Component: Typescript SDK
- [ ] Component: IO connector
- [ ] Component: Beam examples
- [ ] Component: Beam playground
- [ ] Component: Beam katas
- [ ] Component: Website
- [ ] Component: Spark Runner
- [ ] Component: Flink Runner
- [ ] Component: Samza Runner
- [ ] Component: Twister2 Runner
- [ ] Component: Hazelcast Jet Runner
- [X] Component: Google Cloud Dataflow Runner

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.