github/SHARK-Studio
1
1
Fork 0
mirror of https://github.com/nod-ai/SHARK-Studio.git synced 2026-01-09 22:07:55 -05:00
Code Issues Packages Projects Releases Wiki Activity

remove shark 1.0 tests, add support for 2.0 llm

Browse Source 
* add support for external weights

* add tests and edit deps
This commit is contained in:
Daniel Garvey
2023-12-14 21:44:37 -06:00
committed by GitHub
parent f692a012e1
commit ebfcfec338
16 changed files with 377 additions and 576 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
process_skipfiles.py
View File

@@ -8,8 +8,7 @@ from pathlib import Path
# Temporary workaround for transformers/__init__.py.
path_to_transformers_hook = Path(
get_python_lib()
+ "/_pyinstaller_hooks_contrib/hooks/stdhooks/hook-transformers.py"
get_python_lib() + "/_pyinstaller_hooks_contrib/hooks/stdhooks/hook-transformers.py"
)
if path_to_transformers_hook.is_file():
pass
@@ -59,9 +58,7 @@ for line in fileinput.input(path_to_lazy_loader, inplace=True):
# For getting around timm's packaging.
# Refer: https://github.com/pyinstaller/pyinstaller/issues/5673#issuecomment-808731505
path_to_timm_activations = Path(
get_python_lib() + "/timm/layers/activations_jit.py"
)
path_to_timm_activations = Path(get_python_lib() + "/timm/layers/activations_jit.py")
for line in fileinput.input(path_to_timm_activations, inplace=True):
if "@torch.jit.script" in line:
print("@torch.jit._script_if_tracing", end="\n")