diff --git a/examples/tutorial/0_basic.py b/examples/tutorial/0_basic.py index 2cfd8dd..77cc1aa 100644 --- a/examples/tutorial/0_basic.py +++ b/examples/tutorial/0_basic.py @@ -2,7 +2,7 @@ from transformers import AutoModelForSequenceClassification model = AutoModelForSequenceClassification.from_pretrained("facebook/bart-base") # suppose we load BART -from opendelta import Visualization +from bigmodelvis import Visualization print("before modify") Visualization(model).structure_graph() """