Skip to content
This repository has been archived by the owner on Nov 3, 2023. It is now read-only.

[T5] Fix T5 Init Bug #4897

Merged
merged 1 commit into from
Nov 30, 2022
Merged

[T5] Fix T5 Init Bug #4897

merged 1 commit into from
Nov 30, 2022

Conversation

klshuster
Copy link
Contributor

Patch description
Our CI has been failing since earlier versions of HF transformers do not have the torch_dtype option for the T5 model. This patch wraps the init in a try/catch to handle backwards compatibility.

Testing steps
CI

Copy link
Contributor

@mojtaba-komeili mojtaba-komeili left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks for patching.

@klshuster klshuster merged commit 3005c9e into main Nov 30, 2022
@klshuster klshuster deleted the fix_t5 branch November 30, 2022 20:18
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants