diff --git a/apps/stable_diffusion/shark_studio_imports.py b/apps/stable_diffusion/shark_studio_imports.py index 4a55a94bf1..9a3bc4cd28 100644 --- a/apps/stable_diffusion/shark_studio_imports.py +++ b/apps/stable_diffusion/shark_studio_imports.py @@ -15,8 +15,8 @@ # datafiles for pyinstaller datas = [] -datas += collect_data_files("torch") datas += copy_metadata("torch") +datas += copy_metadata("tokenizers") datas += copy_metadata("tqdm") datas += copy_metadata("regex") datas += copy_metadata("requests") @@ -31,18 +31,17 @@ datas += copy_metadata("sentencepiece") datas += copy_metadata("pyyaml") datas += copy_metadata("huggingface-hub") +datas += collect_data_files("torch") datas += collect_data_files("tokenizers") datas += collect_data_files("tiktoken") datas += collect_data_files("accelerate") datas += collect_data_files("diffusers") datas += collect_data_files("transformers") datas += collect_data_files("pytorch_lightning") -datas += collect_data_files("opencv_python") datas += collect_data_files("skimage") datas += collect_data_files("gradio") datas += collect_data_files("gradio_client") datas += collect_data_files("iree") -datas += collect_data_files("google_cloud_storage") datas += collect_data_files("shark", include_py_files=True) datas += collect_data_files("timm", include_py_files=True) datas += collect_data_files("tqdm") @@ -81,3 +80,4 @@ if not any(kw in x for kw in blacklist) ] hiddenimports += [x for x in collect_submodules("iree") if "tests" not in x] +hiddenimports += ["iree._runtime", "iree.compiler._mlir_libs._mlir"]