Spaces:
Running
Running
Ritobrata Ghosh
commited on
Commit
•
5801f13
1
Parent(s):
835ea55
change bart-large-cnn to bart-large in demo folder
Browse files
demo/CustomBARTv4b_model-generate.ipynb
CHANGED
@@ -141,7 +141,7 @@
|
|
141 |
"OUTPUT_VOCAB_SIZE = 16384 + 1 # encoded image token space + 1 for bos\n",
|
142 |
"OUTPUT_LENGTH = 256 + 1 # number of encoded tokens + 1 for bos\n",
|
143 |
"BOS_TOKEN_ID = 16384\n",
|
144 |
-
"BASE_MODEL = 'facebook/bart-large
|
145 |
],
|
146 |
"execution_count": 3,
|
147 |
"outputs": []
|
@@ -563,4 +563,4 @@
|
|
563 |
]
|
564 |
}
|
565 |
]
|
566 |
-
}
|
|
|
141 |
"OUTPUT_VOCAB_SIZE = 16384 + 1 # encoded image token space + 1 for bos\n",
|
142 |
"OUTPUT_LENGTH = 256 + 1 # number of encoded tokens + 1 for bos\n",
|
143 |
"BOS_TOKEN_ID = 16384\n",
|
144 |
+
"BASE_MODEL = 'facebook/bart-large'"
|
145 |
],
|
146 |
"execution_count": 3,
|
147 |
"outputs": []
|
|
|
563 |
]
|
564 |
}
|
565 |
]
|
566 |
+
}
|