Update README.md
Browse files
README.md
CHANGED
@@ -69,7 +69,7 @@ import torch
|
|
69 |
from model import Skip_BART
|
70 |
|
71 |
# Load the model
|
72 |
-
model = Skip_BART.from_pretrained("RS2002/
|
73 |
|
74 |
# Example input
|
75 |
x_encoder = torch.rand((2, 1024, 512))
|
|
|
69 |
from model import Skip_BART
|
70 |
|
71 |
# Load the model
|
72 |
+
model = Skip_BART.from_pretrained("RS2002/Skip-BART")
|
73 |
|
74 |
# Example input
|
75 |
x_encoder = torch.rand((2, 1024, 512))
|