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.

None yet

3 participants