Skip to content

Commit 8265e34

Browse files
authored
Add SHARK SD CLI tool (huggingface#870)
1 parent 8ef8ae0 commit 8265e34

File tree

2 files changed

+77
-0
lines changed

2 files changed

+77
-0
lines changed

.github/workflows/nightly.yml

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -53,6 +53,9 @@ jobs:
5353
pyinstaller web/shark_sd.spec
5454
mv ./dist/shark_sd.exe ./dist/shark_sd_${{ env.package_version_ }}.exe
5555
signtool sign /f C:\shark_2023.cer /csp "eToken Base Cryptographic Provider" /k "${{ secrets.CI_CERT }}" ./dist/shark_sd_${{ env.package_version_ }}.exe
56+
pyinstaller .\shark\examples\shark_inference\stable_diffusion\shark_sd_cli.spec
57+
mv ./dist/shark_sd_cli.exe ./dist/shark_sd_cli_${{ env.package_version_ }}.exe
58+
signtool sign /f C:\shark_2023.cer /csp "eToken Base Cryptographic Provider" /k "${{ secrets.CI_CERT }}" ./dist/shark_sd_cli_${{ env.package_version_ }}.exe
5659
5760
5861
# GHA windows VM OOMs so disable for now
Lines changed: 74 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,74 @@
1+
# -*- mode: python ; coding: utf-8 -*-
2+
from PyInstaller.utils.hooks import collect_data_files
3+
from PyInstaller.utils.hooks import copy_metadata
4+
5+
import sys ; sys.setrecursionlimit(sys.getrecursionlimit() * 5)
6+
7+
datas = []
8+
datas += collect_data_files('torch')
9+
datas += copy_metadata('torch')
10+
datas += copy_metadata('tqdm')
11+
datas += copy_metadata('regex')
12+
datas += copy_metadata('requests')
13+
datas += copy_metadata('packaging')
14+
datas += copy_metadata('filelock')
15+
datas += copy_metadata('numpy')
16+
datas += copy_metadata('tokenizers')
17+
datas += copy_metadata('importlib_metadata')
18+
datas += copy_metadata('torchvision')
19+
datas += copy_metadata('torch-mlir')
20+
datas += copy_metadata('diffusers')
21+
datas += copy_metadata('transformers')
22+
datas += collect_data_files('iree')
23+
datas += collect_data_files('google-cloud-storage')
24+
datas += collect_data_files('shark')
25+
datas += [
26+
( 'resources/prompts.json', 'resources'),
27+
( 'resources/model_db.json', 'resources'),
28+
( 'resources/base_model.json', 'resources'),
29+
( 'resources/opt_flags.json', 'resources'),
30+
]
31+
32+
binaries = []
33+
34+
block_cipher = None
35+
36+
37+
a = Analysis(
38+
['main.py'],
39+
pathex=['.'],
40+
binaries=binaries,
41+
datas=datas,
42+
hiddenimports=['shark', 'shark.*', 'shark.shark_inference', 'shark_inference', 'iree.tools.core' ],
43+
hookspath=[],
44+
hooksconfig={},
45+
runtime_hooks=[],
46+
excludes=[],
47+
win_no_prefer_redirects=False,
48+
win_private_assemblies=False,
49+
cipher=block_cipher,
50+
noarchive=False,
51+
)
52+
pyz = PYZ(a.pure, a.zipped_data, cipher=block_cipher)
53+
54+
exe = EXE(
55+
pyz,
56+
a.scripts,
57+
a.binaries,
58+
a.zipfiles,
59+
a.datas,
60+
[],
61+
name='shark_sd_cli',
62+
debug=False,
63+
bootloader_ignore_signals=False,
64+
strip=False,
65+
upx=True,
66+
upx_exclude=[],
67+
runtime_tmpdir=None,
68+
console=True,
69+
disable_windowed_traceback=False,
70+
argv_emulation=False,
71+
target_arch=None,
72+
codesign_identity=None,
73+
entitlements_file=None,
74+
)

0 commit comments

Comments
 (0)