Fix build script (#93)

Author: jinhongyii

scripts/build_site.sh

@@ -5,6 +5,12 @@ if [[ ! -f $1 ]]; then
     echo "cannot find config" $1
 fi
 
+MLC_LLM_HOME_SET="${MLC_LLM_HOME:-}"
+
+if [ -z ${MLC_LLM_HOME_SET} ]; then
+    export MLC_LLM_HOME="${MLC_LLM_HOME:-mlc-llm}"
+fi
+
 rm -rf site/dist
 mkdir -p site/dist site/_inlcudes
 
@@ -15,9 +21,9 @@ cp web/llm_chat.html site/_includes
 cp web/llm_chat.js site/dist/
 cp web/llm_chat.css site/dist/
 
-cp $MLC_LLM_HOME/dist/tvmjs_runtime.wasi.js site/dist
-cp $MLC_LLM_HOME/dist/tvmjs.bundle.js site/dist
-cp -r $MLC_LLM_HOME/dist/sentencepiece site/dist
+cp dist/tvmjs_runtime.wasi.js site/dist
+cp dist/tvmjs.bundle.js site/dist
+cp -r dist/sentencepiece site/dist
 
 if [ -d "$MLC_LLM_HOME/dist/vicuna-v1-7b-q4f32_0/params" ]; then
     mkdir -p site/dist/vicuna-v1-7b-q4f32_0

site/.gitignore

@@ -1,4 +1,5 @@
 dist
 llm-chat-config.json
+global_config.json
 _includes/stable_diffusion.html
 _site

site/_includes/llm_chat.html

@@ -22,6 +22,13 @@
   tvmjsGlobalEnv.sentencePieceProcessor = getTokenizer;
 </script>
 
+<form>
+  <select id="model-name">
+    <option selected="selected">vicuna-v1-7b-q4f32_0</option>
+    <!-- <option >dolly-v1-3b-q4f32_0</option> -->
+  </select>
+</form>
+
 <script src="dist/llm_chat.js"></script>
 <link href="dist/llm_chat.css" rel="stylesheet" type="text/css"/>