-
Notifications
You must be signed in to change notification settings - Fork 4.3k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[fx]add split module pass and unit test from pipeline passes #1242
[fx]add split module pass and unit test from pipeline passes #1242
Conversation
This reverts commit df7e650.
else: | ||
if len(output_part0) > len(sig.parameters): | ||
output_part0 = output_part0[:len(sig.parameters)] | ||
output_part1 = model_part1(*output_part0) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
HF output may contain multiple values, will output_part1
contain the same number of values?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Yes, it has same number of elements and numerically equal to HF output, but the output_part1 is type of Tuple[torch.Tensor], not HF output class.
No description provided.