├── .gitignore ├── .gitmodules ├── LICENSE ├── README.md ├── convert_model_from_pth_safetensors.py ├── deploy.py ├── log_db ├── database_tuning_record.json └── database_workload.json ├── log_db_my_clip_unet ├── database_tuning_record.json └── database_workload.json ├── log_db_my_unet_softmax2 ├── database_tuning_record.json └── database_workload.json ├── log_db_my_vae ├── database_tuning_record.json └── database_workload.json ├── requirements.txt ├── scripts ├── build_site.sh ├── gh_deploy_site.sh ├── local_deploy_site.sh ├── prep_deps.sh └── rpc_debug_deploy.sh ├── setup.py ├── site ├── .gitignore ├── _config.yml ├── _includes │ └── head.html ├── img │ ├── fig │ │ ├── browser-screenshot.png │ │ ├── pipeline.svg │ │ └── workflow.svg │ ├── gif │ │ ├── direct-run.gif │ │ └── intermediate-render.gif │ └── logo │ │ ├── catalyst.svg │ │ ├── cmuscs.png │ │ ├── mlc-logo-with-text-landscape.svg │ │ └── octoml.png └── index.md ├── tests └── debug │ ├── compare_lib.py │ └── rpc_run.py ├── v1-inference.yaml ├── web ├── gh-page-config.json ├── local-config.json ├── stable_diffusion.html └── stable_diffusion.js └── web_stable_diffusion ├── __init__.py ├── models ├── activations.py ├── attention.py ├── attention_processor.py ├── autoencoder_kl.py ├── embeddings.py ├── resnet.py ├── transformer_2d.py ├── unet_2d_blocks.py ├── unet_2d_condition.py ├── unet_blocks.py └── vae.py ├── rpc_testing.py ├── runtime ├── __init__.py └── scheduler_runtime.py ├── trace ├── __init__.py ├── model_trace.py └── scheduler_trace.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/.gitmodules -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/README.md -------------------------------------------------------------------------------- /convert_model_from_pth_safetensors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/convert_model_from_pth_safetensors.py -------------------------------------------------------------------------------- /deploy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/deploy.py -------------------------------------------------------------------------------- /log_db/database_tuning_record.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/log_db/database_tuning_record.json -------------------------------------------------------------------------------- /log_db/database_workload.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/log_db/database_workload.json -------------------------------------------------------------------------------- /log_db_my_clip_unet/database_tuning_record.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/log_db_my_clip_unet/database_tuning_record.json -------------------------------------------------------------------------------- /log_db_my_clip_unet/database_workload.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/log_db_my_clip_unet/database_workload.json -------------------------------------------------------------------------------- /log_db_my_unet_softmax2/database_tuning_record.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/log_db_my_unet_softmax2/database_tuning_record.json -------------------------------------------------------------------------------- /log_db_my_unet_softmax2/database_workload.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/log_db_my_unet_softmax2/database_workload.json -------------------------------------------------------------------------------- /log_db_my_vae/database_tuning_record.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/log_db_my_vae/database_tuning_record.json -------------------------------------------------------------------------------- /log_db_my_vae/database_workload.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/log_db_my_vae/database_workload.json -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/build_site.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/scripts/build_site.sh -------------------------------------------------------------------------------- /scripts/gh_deploy_site.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/scripts/gh_deploy_site.sh -------------------------------------------------------------------------------- /scripts/local_deploy_site.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/scripts/local_deploy_site.sh -------------------------------------------------------------------------------- /scripts/prep_deps.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/scripts/prep_deps.sh -------------------------------------------------------------------------------- /scripts/rpc_debug_deploy.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/scripts/rpc_debug_deploy.sh -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/setup.py -------------------------------------------------------------------------------- /site/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/site/.gitignore -------------------------------------------------------------------------------- /site/_config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/site/_config.yml -------------------------------------------------------------------------------- /site/_includes/head.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/site/_includes/head.html -------------------------------------------------------------------------------- /site/img/fig/browser-screenshot.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/site/img/fig/browser-screenshot.png -------------------------------------------------------------------------------- /site/img/fig/pipeline.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/site/img/fig/pipeline.svg -------------------------------------------------------------------------------- /site/img/fig/workflow.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/site/img/fig/workflow.svg -------------------------------------------------------------------------------- /site/img/gif/direct-run.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/site/img/gif/direct-run.gif -------------------------------------------------------------------------------- /site/img/gif/intermediate-render.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/site/img/gif/intermediate-render.gif -------------------------------------------------------------------------------- /site/img/logo/catalyst.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/site/img/logo/catalyst.svg -------------------------------------------------------------------------------- /site/img/logo/cmuscs.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/site/img/logo/cmuscs.png -------------------------------------------------------------------------------- /site/img/logo/mlc-logo-with-text-landscape.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/site/img/logo/mlc-logo-with-text-landscape.svg -------------------------------------------------------------------------------- /site/img/logo/octoml.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/site/img/logo/octoml.png -------------------------------------------------------------------------------- /site/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/site/index.md -------------------------------------------------------------------------------- /tests/debug/compare_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/tests/debug/compare_lib.py -------------------------------------------------------------------------------- /tests/debug/rpc_run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/tests/debug/rpc_run.py -------------------------------------------------------------------------------- /v1-inference.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/v1-inference.yaml -------------------------------------------------------------------------------- /web/gh-page-config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/web/gh-page-config.json -------------------------------------------------------------------------------- /web/local-config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/web/local-config.json -------------------------------------------------------------------------------- /web/stable_diffusion.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/web/stable_diffusion.html -------------------------------------------------------------------------------- /web/stable_diffusion.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/web/stable_diffusion.js -------------------------------------------------------------------------------- /web_stable_diffusion/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/web_stable_diffusion/__init__.py -------------------------------------------------------------------------------- /web_stable_diffusion/models/activations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/web_stable_diffusion/models/activations.py -------------------------------------------------------------------------------- /web_stable_diffusion/models/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/web_stable_diffusion/models/attention.py -------------------------------------------------------------------------------- /web_stable_diffusion/models/attention_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/web_stable_diffusion/models/attention_processor.py -------------------------------------------------------------------------------- /web_stable_diffusion/models/autoencoder_kl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/web_stable_diffusion/models/autoencoder_kl.py -------------------------------------------------------------------------------- /web_stable_diffusion/models/embeddings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/web_stable_diffusion/models/embeddings.py -------------------------------------------------------------------------------- /web_stable_diffusion/models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/web_stable_diffusion/models/resnet.py -------------------------------------------------------------------------------- /web_stable_diffusion/models/transformer_2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/web_stable_diffusion/models/transformer_2d.py -------------------------------------------------------------------------------- /web_stable_diffusion/models/unet_2d_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/web_stable_diffusion/models/unet_2d_blocks.py -------------------------------------------------------------------------------- /web_stable_diffusion/models/unet_2d_condition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/web_stable_diffusion/models/unet_2d_condition.py -------------------------------------------------------------------------------- /web_stable_diffusion/models/unet_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/web_stable_diffusion/models/unet_blocks.py -------------------------------------------------------------------------------- /web_stable_diffusion/models/vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/web_stable_diffusion/models/vae.py -------------------------------------------------------------------------------- /web_stable_diffusion/rpc_testing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/web_stable_diffusion/rpc_testing.py -------------------------------------------------------------------------------- /web_stable_diffusion/runtime/__init__.py: -------------------------------------------------------------------------------- 1 | from .scheduler_runtime import * 2 | -------------------------------------------------------------------------------- /web_stable_diffusion/runtime/scheduler_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/web_stable_diffusion/runtime/scheduler_runtime.py -------------------------------------------------------------------------------- /web_stable_diffusion/trace/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/web_stable_diffusion/trace/__init__.py -------------------------------------------------------------------------------- /web_stable_diffusion/trace/model_trace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/web_stable_diffusion/trace/model_trace.py -------------------------------------------------------------------------------- /web_stable_diffusion/trace/scheduler_trace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/web_stable_diffusion/trace/scheduler_trace.py -------------------------------------------------------------------------------- /web_stable_diffusion/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/happyme531/RK3588-stable-diffusion-GPU/HEAD/web_stable_diffusion/utils.py --------------------------------------------------------------------------------