mirror of
https://github.com/nod-ai/SHARK-Studio.git
synced 2026-01-08 21:38:04 -05:00
f6d41affd9
* Dan shark studio (#1970) * Fix issue in Falcon-GPTQ * initial webui and llama2 --------- Co-authored-by: Vivek Khandelwal <vivekkhandelwal1424@gmail.com> * Fix formatting. --------- Co-authored-by: Daniel Garvey <34486624+dan-garvey@users.noreply.github.com> Co-authored-by: Vivek Khandelwal <vivekkhandelwal1424@gmail.com>
53 lines
1.1 KiB
Plaintext
53 lines
1.1 KiB
Plaintext
-f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
|
|
--pre
|
|
|
|
setuptools
|
|
wheel
|
|
|
|
# SHARK Runner
|
|
tqdm
|
|
|
|
# SHARK Downloader
|
|
google-cloud-storage
|
|
|
|
# Testing
|
|
pytest
|
|
pytest-xdist
|
|
pytest-forked
|
|
Pillow
|
|
parameterized
|
|
|
|
#shark-turbine @ git+https://github.com/nod-ai/SHARK-Turbine.git@main
|
|
# Add transformers, diffusers and scipy since it most commonly used
|
|
tokenizers==0.13.3
|
|
transformers
|
|
diffusers
|
|
#accelerate is now required for diffusers import from ckpt.
|
|
accelerate
|
|
scipy
|
|
ftfy
|
|
gradio==3.44.3
|
|
altair
|
|
omegaconf
|
|
# 0.3.2 doesn't have binaries for arm64
|
|
safetensors==0.3.1
|
|
opencv-python
|
|
scikit-image
|
|
pytorch_lightning # for runwayml models
|
|
tk
|
|
pywebview
|
|
sentencepiece
|
|
py-cpuinfo
|
|
tiktoken # for codegen
|
|
joblib # for langchain
|
|
timm # for MiniGPT4
|
|
langchain
|
|
einops # for zoedepth
|
|
pydantic==2.4.1 # pin until pyinstaller-hooks-contrib works with beta versions
|
|
|
|
# Keep PyInstaller at the end. Sometimes Windows Defender flags it but most folks can continue even if it errors
|
|
pefile
|
|
pyinstaller
|
|
|
|
# vicuna quantization
|
|
brevitas @ git+https://github.com/Xilinx/brevitas.git@56edf56a3115d5ac04f19837b388fd7d3b1ff7ea |