This commit is contained in:
shengdinghu 2023-03-11 09:26:45 +00:00
parent 2c8de84ad8
commit cfde7552f1
1 changed files with 1 additions and 1 deletions

View File

@ -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()
"""