RS2002 commited on
Commit
3988493
·
verified ·
1 Parent(s): 83fc46d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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/Skip_BART")
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))