TransformerLensOrg / TransformerLens

A library for mechanistic interpretability of GPT-style language models
https://transformerlensorg.github.io/TransformerLens/
MIT License
1.17k stars 241 forks source link

Lock datasets version #632

Closed courtney-sims closed 3 weeks ago

courtney-sims commented 3 weeks ago

Description

This change resolves dependency errors I encountered when running the setup for Interactive_Neuroscope colab.:

The datasets package released version 2.19.2 on June 3 and this version causes dependency issues with other requirements, like google-colab 1.0.0 (which does not have a newer version). Pinning datasets to the previous 2.19.1 resolved the issue and allowed the setup to complete without error.

Type of change

Bug fix (non-breaking change which fixes an issue)

Screenshots

Please attach before and after screenshots of the change if applicable.

Before After
Screenshot 2024-06-08 at 11 51 14 AM Screenshot 2024-06-08 at 11 54 58 AM
Screenshot 2024-06-08 at 11 51 14 AM ^^^

Checklist:

bryce13950 commented 3 weeks ago

Thanks!