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

Remove done TODOs, a sup PR for #1644 (#1647)

Browse Source
This commit is contained in:
jinchen62
2023-07-12 23:30:45 -07:00
committed by GitHub
parent a909fc2e78
commit 91027f8719
1 changed files with 0 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
apps/language_models/scripts/vicuna.py
View File

@@ -574,7 +574,6 @@ class ShardedVicuna(SharkLLMBase):
use_tracing=False,
verbose=False,
)
# TODO: apply --canonicalize to unpack tensor for int4
print(f"[DEBUG] converting torch to linalg")
run_pipeline_with_repro_report(
module0,
@@ -619,7 +618,6 @@ class ShardedVicuna(SharkLLMBase):
use_tracing=False,
verbose=False,
)
# TODO: apply --canonicalize to unpack tensor for int4
print(f"[DEBUG] converting torch to linalg")
run_pipeline_with_repro_report(
module1,
@@ -1018,7 +1016,6 @@ class UnshardedVicuna(SharkLLMBase):
use_tracing=False,
verbose=False,
)
# TODO: apply --canonicalize to unpack tensor for int4
print(f"[DEBUG] converting torch to linalg")
run_pipeline_with_repro_report(
module,
@@ -1187,7 +1184,6 @@ class UnshardedVicuna(SharkLLMBase):
use_tracing=False,
verbose=False,
)
# TODO: apply --canonicalize to unpack tensor for int4
print(f"[DEBUG] converting torch to linalg")
run_pipeline_with_repro_report(
module,