fixed #52
This commit is contained in:
parent
2c8de84ad8
commit
cfde7552f1
|
@ -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()
|
||||
"""
|
||||
|
|
Loading…
Reference in New Issue