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

fixed issue where a file wouldn't be downloaded when running full bloom

Browse Source
This commit is contained in:
Elias Joseph
2023-03-22 19:21:22 +00:00
parent f1fb363403
commit 2f891a6c23
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
shark/examples/shark_inference/sharded_bloom.py
View File

@@ -742,6 +742,9 @@ if __name__ == "__main__":
print(
"WARNING: It is not advised to turn on both download and create_mlirs"
)
if args.model_name == "bloom" and (args.create_mlirs or args.download):
urllib.request.urlretrieve("https://huggingface.co/bigscience/bloom/resolve/main/pytorch_model_00001-of-00072.bin", f"{args.model_path}/pytorch_model_00001-of-00072.bin")
if args.download:
download_model(args.model_path, args.model_name)
if args.create_mlirs: