Make fx traced model with the use of `past_key_values` pickable again?

[michaelbenayoun]

With the support of the transformers.cache_utils.Cache classes when doing symbolic trace (#30259), it is not possible to pickle a traced model which uses past_key_values, should we work to add support for that?

[github-actions]

This issue has been automatically marked as stale because it has not had recent activity. If you think this still needs to be addressed please comment on this thread.

Please note that issues that do not follow the contributing guidelines are likely to be ignored.

[LysandreJik]

cc @ArthurZucker @gante

[michaelbenayoun]

Created this issue to have the discussion but I do not know if we actually want to support that.

[github-actions]

This issue has been automatically marked as stale because it has not had recent activity. If you think this still needs to be addressed please comment on this thread.

Please note that issues that do not follow the contributing guidelines are likely to be ignored.

[gante]

@michaelbenayoun sorry, this was lost in the notification queue 😬

I think the PT team is taking care of what you asked in this PR: #32253