mirror of
https://github.com/nod-ai/SHARK-Studio.git
synced 2026-01-09 13:57:54 -05:00
d2c3752dc7
* Fix batch count * Add button to unload models manually. * Add compiled pipeline option * Add brevitas to requirements * Tweaks to chatbot * Change script loading trigger
46 lines
1.1 KiB
Plaintext
46 lines
1.1 KiB
Plaintext
-f https://download.pytorch.org/whl/nightly/cpu
|
|
-f https://iree.dev/pip-release-links.html
|
|
--pre
|
|
|
|
setuptools
|
|
wheel
|
|
|
|
|
|
torch==2.3.0
|
|
shark-turbine @ git+https://github.com/iree-org/iree-turbine.git@main
|
|
turbine-models @ git+https://github.com/nod-ai/SHARK-Turbine.git@ean-unify-sd#subdirectory=models
|
|
diffusers @ git+https://github.com/nod-ai/diffusers@v0.24.0-release
|
|
brevitas @ git+https://github.com/Xilinx/brevitas.git@6695e8df7f6a2c7715b9ed69c4b78157376bb60b
|
|
|
|
# SHARK Runner
|
|
tqdm
|
|
|
|
# SHARK Downloader
|
|
google-cloud-storage
|
|
|
|
# Testing
|
|
pytest
|
|
Pillow
|
|
parameterized
|
|
|
|
# Add transformers, diffusers and scipy since it most commonly used
|
|
#accelerate is now required for diffusers import from ckpt.
|
|
accelerate
|
|
scipy
|
|
transformers==4.37.1
|
|
torchsde # Required for Stable Diffusion SDE schedulers.
|
|
ftfy
|
|
gradio==4.29.0
|
|
altair
|
|
omegaconf
|
|
# 0.3.2 doesn't have binaries for arm64
|
|
safetensors==0.3.1
|
|
py-cpuinfo
|
|
pydantic==2.4.1 # pin until pyinstaller-hooks-contrib works with beta versions
|
|
mpmath==1.3.0
|
|
optimum
|
|
|
|
# Keep PyInstaller at the end. Sometimes Windows Defender flags it but most folks can continue even if it errors
|
|
pefile
|
|
pyinstaller
|