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

Add support for ROCM (Windows) in Studio + compile utils (#1770)

Browse Source 
* WIP: MSVC ROCM support for SHARK Studio

* Make get_iree_rocm_args platform-agnostic.

* Update stable_args.py

* Update rocm arg handling in SD utils

* Guard quantization imports.

Co-authored-by: jam https://github.com/jammm
This commit is contained in:
Ean Garvey
2023-08-25 22:56:05 -05:00
committed by GitHub
parent 9e348a114e
commit 9c8cbaf498
12 changed files with 119 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
process_skipfiles.py
View File

@@ -6,7 +6,7 @@ from distutils.sysconfig import get_python_lib
import fileinput
from pathlib import Path
# Temorary workaround for transformers/__init__.py.
# Temporary workaround for transformers/__init__.py.
path_to_transformers_hook = Path(
get_python_lib()
+ "/_pyinstaller_hooks_contrib/hooks/stdhooks/hook-transformers.py"