Cleanup some local script

.gitignore

@@ -1,7 +1,7 @@
 dist/
 params/
 **/*.png
-
+*.bak
 # Byte-compiled / optimized / DLL files
 __pycache__/
 *.py[cod]
@@ -315,3 +315,5 @@ gallery/how_to/work_with_microtvm/micro_tvmc.py
 
 # GDB history file
 .gdb_history
+
+3rdparty

web/local_deploy.sh → scripts/build_site.sh

@@ -1,28 +1,24 @@
 #!/bin/bash
 set -euxo pipefail
-PROJ_ROOT=`pwd`
+
+if [[ ! -f $1 ]]; then
+    echo "cannot find config" $1
+fi
 
 rm -rf site/dist
 mkdir -p site/dist site/_inlcudes
 
 echo "Copy local configurations.."
-cp web/local-config.json site/stable-diffusion-config.json
-
+cp $1 site/stable-diffusion-config.json
 echo "Copy files..."
 cp web/stable_diffusion.html site/_includes
 cp web/stable_diffusion.js site/dist
 
-
 cp dist/scheduler_consts.json site/dist
 cp dist/stable_diffusion_webgpu.wasm site/dist
 
 cp dist/tvmjs_runtime.wasi.js site/dist
 cp dist/tvmjs.bundle.js site/dist
 cp -r dist/tokenizers-wasm site/dist
 
-cd site && jekyll b
-
-echo "symlink parameter location to site.."
-ln -s ${PROJ_ROOT}/dist/params _site/dist/webgpu-sd-v1-5
-
-jekyll serve  --skip-initial-build  --baseurl /web-stable-diffusion --port 8888
+cd site && jekyll b && cd ..

scripts/gh_deploy_site.sh

@@ -0,0 +1,17 @@
+#!/bin/bash
+set -euxo pipefail
+
+scripts/build_site.sh dist/gh-page-config.json
+
+git fetch
+git checkout -B gh-pages origin/gh-pages
+rm -rf docs .gitignore
+mkdir -p docs
+cp -rf site/_site/* docs
+touch docs/.nojekyll
+
+DATE=`date`
+git add docs && git commit -am "Build at ${DATE}"
+git push origin gh-pages
+git checkout main && git submodule update
+echo "Finish deployment at ${DATE}"

scripts/local_deploy_site.sh

@@ -0,0 +1,9 @@
+#!/bin/bash
+set -euxo pipefail
+
+scripts/build_site.sh web/local-config.json
+
+echo "symlink parameter location to site.."
+
+ln -s `pwd`/dist/params site/_site/dist/webgpu-sd-v1-5
+cd site && jekyll serve  --skip-initial-build --host localhost --baseurl /web-stable-diffusion --port 8888

scripts/prep_deps.sh

@@ -0,0 +1,29 @@
+#!/bin/bash
+# This file prepares all the necessary dependencies for the web build.
+set -euxo pipefail
+
+emcc --version
+npm --version
+cargo --version
+
+TVM_HOME_SET="${TVM_HOME:-}"
+
+if [[ -z ${TVM_HOME_SET} ]]; then
+    if [[ ! -d "3rdparty/tvm" ]]; then
+        echo "Do not find TVM_HOME env variable, cloning a version as source".
+        git clone https://github.com/apache/tvm 3rdparty/tvm --branch unity --recursive
+    fi
+    export TVM_HOME="${TVM_HOME:-3rdparty/tvm}"
+fi
+
+export TOKENZIER_WASM_HOME="3rdparty/tokenizers-wasm"
+
+mkdir -p dist
+cd ${TVM_HOME}/web && make && npm install && npm run bundle && cd -
+git submodule update --init
+cd ${TOKENZIER_WASM_HOME} && wasm-pack build --target web && cd -
+rm -rf dist/tokenizers-wasm
+cp -r ${TOKENZIER_WASM_HOME}/pkg dist/tokenizers-wasm
+
+echo "Exporting tvmjs runtime dist files"
+python -c "from tvm.contrib import tvmjs; tvmjs.export_runtime(\"dist\")"

setup.py

@@ -0,0 +1,29 @@
+from distutils.core import setup
+from setuptools import find_packages
+
+# Note there is no need to setup when
+# running locally in this folder.
+
+setup(
+    name = "web-stable-diffusion",
+    version = "0.1.0",
+    license="Apache-2.0",
+    description = "Stable diffusion on browser",
+    author = "Web stable diffusion contributors",
+    url = "https://github.com/mlc-ai/web-stable-diffusion",
+    keywords = [],
+    packages = find_packages(),
+    install_requires = [
+        "torch",
+        "diffusers",
+        "transformers"
+    ],
+    classifiers = [
+        "Development Status :: 3 - Alpha",
+        "Intended Audience :: Developers",
+        "Topic :: Software Development :: Build Tools",
+        "License :: OSI Approved :: Apache Software License",
+        "Programming Language :: Python :: 3",
+        "Programming Language :: Python :: 3.6",
+  ],
+)

site/_config.yml

@@ -1,6 +1,6 @@
 
 name: "Web Stable Diffusion"
-short_name: "WiSe"
+short_name: "WebSD"
 
 url: https://mlc.ai/web-stable-diffusion
 

site/index.md

@@ -6,7 +6,4 @@ notitle: true
 
 # Web Stable Diffusion
 
-
-This is a page
-
 {% include stable_diffusion.html %}

web/stable_diffusion.html

@@ -30,7 +30,7 @@ <h2> Image Generation Demo </h2>
   id="inputPrompt"
   type="text"
   value="A photo of an astronaut riding a horse on mars"
-  size="77"/>
+  size="77"/> <br>
   Negative Prompt(can be left empty) <input
   name="negativePrompt"
   id="negativePrompt"