fixed a bug in saving

This commit is contained in:
Yousef Rafat 2025-07-09 00:38:15 +03:00
parent 84bcc096ce
commit 3ac2861b17

View File

@ -183,8 +183,9 @@ class Hunyuan3DDiTFlowMatchingPipeline(nn.Module):
try: try:
if save_file is not None: if save_file is not None:
for output in mesh: for i, output in enumerate(mesh):
save_glb(output.mesh_v, output.mesh_f, save_file, numpy_ready = True) output_file = f"{save_file}_{i}" if len(mesh) > 1 else save_file
save_glb(output.mesh_v, output.mesh_f, output_file, numpy_ready = True)
except Exception as e: except Exception as e:
print(e) print(e)