Update generate_sharktank.py

This commit is contained in:
Ean Garvey
2023-11-29 23:28:13 -06:00
committed by GitHub
parent fe80a1a17d
commit 665437c999

View File

@@ -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