Skip to content

Fix BERT model loading by removing deprecated cache_dir parameter#12

Open
Hrushik777 wants to merge 2 commits intoOpenNyAI:masterfrom
Hrushik777:master
Open

Fix BERT model loading by removing deprecated cache_dir parameter#12
Hrushik777 wants to merge 2 commits intoOpenNyAI:masterfrom
Hrushik777:master

Conversation

@Hrushik777
Copy link
Copy Markdown

Removed deprecated cache_dir parameter from BertModel.from_pretrained().
This parameter is no longer required in newer HuggingFace versions and causes loading issues.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant