From 8a39d5414e536d3ff5c3cde1fae71f604d1b3762 Mon Sep 17 00:00:00 2001
From: Ines Montani <ines@ines.io>
Date: Mon, 5 Oct 2020 21:43:51 +0200
Subject: [PATCH] Update quickstart [ci skip]

---
 website/src/widgets/quickstart-install.js | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/website/src/widgets/quickstart-install.js b/website/src/widgets/quickstart-install.js
index a8bdf21dc..ab91b8e30 100644
--- a/website/src/widgets/quickstart-install.js
+++ b/website/src/widgets/quickstart-install.js
@@ -17,6 +17,7 @@ const CUDA = {
     '10.1': 'cuda101',
     '10.2': 'cuda102',
 }
+const LANG_EXTRAS = ['zh', 'ja'] // only for languages with models
 const DATA = [
     {
         id: 'os',
@@ -81,7 +82,13 @@ const QuickstartInstall = ({ id, title }) => {
     const showDropdown = {
         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)
         .join(',')
     return (