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") model = AutoModelForSequenceClassification.from_pretrained("facebook/bart-base")
# suppose we load BART # suppose we load BART
from opendelta import Visualization from bigmodelvis import Visualization
print("before modify") print("before modify")
Visualization(model).structure_graph() Visualization(model).structure_graph()
""" """