diffusion: skip _save_gt_output for 3d/mesh (#32117)
This commit is contained in:
@@ -841,6 +841,10 @@ Pinned revision used by this check: {SGL_TEST_FILES_CI_DATA_REVISION}
|
|||||||
num_gpus = case.server_args.num_gpus
|
num_gpus = case.server_args.num_gpus
|
||||||
is_video = case.server_args.modality == "video"
|
is_video = case.server_args.modality == "video"
|
||||||
|
|
||||||
|
if case.server_args.modality == "3d":
|
||||||
|
logger.info("Skipping GT save for mesh (3d) case: %s", case.id)
|
||||||
|
return
|
||||||
|
|
||||||
if case.server_args.modality == "action":
|
if case.server_args.modality == "action":
|
||||||
output_path = out_dir / f"{case.id}_{num_gpus}gpu.json"
|
output_path = out_dir / f"{case.id}_{num_gpus}gpu.json"
|
||||||
output_path.write_bytes(content)
|
output_path.write_bytes(content)
|
||||||
|
|||||||
Reference in New Issue
Block a user