Skip to content

Conversation

@trueroyce
Copy link

Purpose

This PR adds a new example notebook to demonstrate the use of the RetNet model and updates the codebase to ensure compatibility with the latest dependencies.

Changes

  • Added news_fake_example.ipynb to showcase how to load and test the RetNet model with a sample dataset.
  • Updated retnet/modeling_retnet.py to fix compatibility issues with updated library versions.

Notes

This PR addresses compatibility issues caused by updates in the transformers library. Specifically, the top_k_top_p_filtering function is now imported from transformers.generation.utils. This change ensures the codebase remains functional with the latest version of transformers.

@review-notebook-app
Copy link

Check out this pull request on  ReviewNB

See visual diffs & provide feedback on Jupyter Notebooks.


Powered by ReviewNB

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