mirror of
https://github.com/nod-ai/AMD-SHARK-Studio.git
synced 2026-04-25 03:00:12 -04:00
Update generate_sharktank.py
This commit is contained in:
@@ -47,7 +47,7 @@ def save_torch_model(torch_model_list, local_tank_cache, import_args):
|
||||
is_dynamic = row[3]
|
||||
mlir_type = row[4]
|
||||
is_decompose = row[5]
|
||||
tracing_required = False if tracing_required == "False" else True
|
||||
tracing_required = True
|
||||
is_dynamic = False
|
||||
print("generating artifacts for: " + torch_model_name)
|
||||
model = None
|
||||
|
||||
Reference in New Issue
Block a user