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))
|