diff --git a/exe/win10.directml.dir.spec b/exe/win10.directml.dir.spec index 1b1f89da..507972d7 100644 --- a/exe/win10.directml.dir.spec +++ b/exe/win10.directml.dir.spec @@ -13,7 +13,6 @@ dynamic_packages = [ "filelock", "huggingface-hub", "numpy", - "onnxruntime", "onnxruntime-directml", "packaging", "pyyaml", @@ -35,6 +34,13 @@ datas = [ "models/**", "utils/**", ]), + collect_data_files("codeformer", include_py_files=True, includes=[ + "basicsr/archs/**", + "basicsr/data/**", + "basicsr/losses/**", + "basicsr/models/**", + "basicsr/utils/**", + ]), collect_data_files("realesrgan", include_py_files=True, includes=[ "archs/**", "data/**", @@ -46,6 +52,9 @@ datas = [ "transformers/**", "tools/**", ]), + collect_data_files("timm", include_py_files=True, includes=[ + "models/**", + ]), collect_data_files("transformers", include_py_files=True, includes=[ "**", ]), diff --git a/exe/win10.directml.file.spec b/exe/win10.directml.file.spec index 658abac3..f26e846a 100644 --- a/exe/win10.directml.file.spec +++ b/exe/win10.directml.file.spec @@ -13,7 +13,6 @@ dynamic_packages = [ "filelock", "huggingface-hub", "numpy", - "onnxruntime", "onnxruntime-directml", "packaging", "pyyaml", @@ -35,6 +34,13 @@ datas = [ "models/**", "utils/**", ]), + collect_data_files("codeformer", include_py_files=True, includes=[ + "basicsr/archs/**", + "basicsr/data/**", + "basicsr/losses/**", + "basicsr/models/**", + "basicsr/utils/**", + ]), collect_data_files("realesrgan", include_py_files=True, includes=[ "archs/**", "data/**", @@ -46,6 +52,9 @@ datas = [ "transformers/**", "tools/**", ]), + collect_data_files("timm", include_py_files=True, includes=[ + "models/**", + ]), collect_data_files("transformers", include_py_files=True, includes=[ "**", ]),