github/AMD-SHARK-Studio
1
1
Fork 0
mirror of https://github.com/nod-ai/AMD-SHARK-Studio.git synced 2026-04-03 03:00:17 -04:00
Code Issues Packages Projects Releases Wiki Activity

fix missing model download path (#281)

Browse Source
This commit is contained in:
Daniel Garvey
2022-08-17 23:02:50 -05:00
committed by GitHub
parent 55bcb2eb3c
commit 82c541dfb8
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
shark/shark_downloader.py
View File

@@ -153,7 +153,8 @@ def download_tflite_model(model_name, dynamic=False):
model_dir = os.path.join(WORKDIR, model_dir_name)
local_hash = str(np.load(os.path.join(model_dir, "hash.npy")))
gs_hash = (
'gsutil -o "GSUtil:parallel_process_count=1" cp gs://shark_tank'
'gsutil -o "GSUtil:parallel_process_count=1" cp gs://shark_tank/'
+ shark_default_sha
+ "/"
+ model_dir_name
+ "/hash.npy"