Update quickstart [ci skip]

This commit is contained in:
Ines Montani 2020-10-05 21:43:51 +02:00
parent 2d0c0134bc
commit 8a39d5414e

View File

@ -17,6 +17,7 @@ const CUDA = {
'10.1': 'cuda101', '10.1': 'cuda101',
'10.2': 'cuda102', '10.2': 'cuda102',
} }
const LANG_EXTRAS = ['zh', 'ja'] // only for languages with models
const DATA = [ const DATA = [
{ {
id: 'os', id: 'os',
@ -81,7 +82,13 @@ const QuickstartInstall = ({ id, title }) => {
const showDropdown = { const showDropdown = {
hardware: () => hardware === 'gpu', hardware: () => hardware === 'gpu',
} }
const pipExtras = [hardware === 'gpu' && cuda, train && 'transformers', train && 'lookups'] const modelExtras = train ? selectedModels.filter(m => LANG_EXTRAS.includes(m)) : []
const pipExtras = [
hardware === 'gpu' && cuda,
train && 'transformers',
train && 'lookups',
...modelExtras,
]
.filter(e => e) .filter(e => e)
.join(',') .join(',')
return ( return (