You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
## Description
Implements #928
Users want the first pipeline stage to accept kwargs if the original
program does.
This is controlled by the `_codegen` field of the graph as @angelayi
suggests, so we make a copy from the traced program to submod0.
## Feature/Issue validation/testing
Added kwargs in test_fwd.py.
Also changed a few HF examples to directly kwargs.
People would like to pass kwargs to model as in non-pipeline runs, instead of converting it to args tuple first.
See: huggingface/accelerate#2345
PiPPy used to support kwargs before tracer migration. We'd need to add it back.
The text was updated successfully, but these errors were encountered: