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")
|
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()
|
||||||
"""
|
"""
|
||||||
|
|
Loading…
Reference in New Issue