fix library imports

stanfordnlp · Jan 12, 2024 · e78348f · e78348f
1 parent f5b298a
commit e78348f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -2,9 +2,8 @@ torch
 transformers
 datasets
 notebook
-protobuf==3.19.1
+protobuf==3.20.*
 matplotlib
-wandb
 seaborn
 ipywidgets
 plotnine

diff --git a/setup.py b/setup.py
@@ -10,7 +10,7 @@
 
 setup(
     name="pyvene",
-    version="0.0.3",
+    version="0.0.4",
     description="Use Activation Intervention to Interpret Causal Mechanism of Model",
     long_description=long_description,
     long_description_content_type='text/markdown',