fix(exe): include realesrgan and coloredlogs in bundles

ssube · Mar 31, 2023 · 9b05b6b · 9b05b6b
1 parent ee3774b
commit 9b05b6b
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/exe/win10.directml.dir.spec b/exe/win10.directml.dir.spec
@@ -40,6 +40,7 @@ datas = [
     ]),
     collect_data_files("onnxruntime", include_py_files=True, includes=[
         "transformers/**",
+        "tools/**",
     ]),
     collect_data_files("transformers", include_py_files=True, includes=[
         "**",
@@ -55,7 +56,7 @@ a = Analysis(
         *metadatas,
         *datas,
     ],
-    hiddenimports=['onnxruntime', 'onnxruntime-directml', 'tqdm'],
+    hiddenimports=['coloredlogs', 'onnxruntime', 'onnxruntime-directml', 'tqdm'],
     hookspath=[],
     hooksconfig={},
     runtime_hooks=[],

diff --git a/exe/win10.directml.file.spec b/exe/win10.directml.file.spec
@@ -31,6 +31,17 @@ datas = [
         "models/**",
         "utils/**",
     ]),
+    collect_data_files("realesrgan", include_py_files=True, includes=[
+        "archs/**",
+        "data/**",
+        "losses/**",
+        "models/**",
+        "utils/**",
+    ]),
+    collect_data_files("onnxruntime", include_py_files=True, includes=[
+        "transformers/**",
+        "tools/**",
+    ]),
     collect_data_files("transformers", include_py_files=True, includes=[
         "**",
     ]),
@@ -45,7 +56,7 @@ a = Analysis(
         *metadatas,
         *datas,
     ],
-    hiddenimports=['onnxruntime', 'onnxruntime-directml', 'tqdm'],
+    hiddenimports=['coloredlogs', 'onnxruntime', 'onnxruntime-directml', 'tqdm'],
     hookspath=[],
     hooksconfig={},
     runtime_hooks=[],