Aero-Ex

/

Trellis2-GGUF

like 17

Getting the following error when trying to generate a mesh. The process runs fine until the decode step, then fails with NameError: name 'tqdm' is not defined.

Using Modly with Trellis2 GGUF extension on Windows 11, RTX 50 series GPU.

Error: Traceback (most recent call last):
File "C:\Program Files\Modly\resources\api\routers\generation.py", line 153, in _run_generation
output_path = await loop.run_in_executor(
^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\Program Files\Modly\resources\python-embed\Lib\concurrent\futures\thread.py", line 58, in run
result = self.fn(*self.args, **self.kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\Program Files\Modly\resources\api\routers\generation.py", line 155, in
lambda: gen.generate(image_bytes, params, progress_cb, cancel_event)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\Program Files\Modly\resources\api\services\extension_process.py", line 232, in generate
raise RuntimeError(msg.get("traceback") or msg.get("message", "Unknown error"))
RuntimeError: Traceback (most recent call last):
File "C:\Program Files\Modly\resources\api\runner.py", line 160, in main
output_path = gen.generate(image_bytes, params, progress_cb, cancel_evt)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\Users\MSI\Documents\Modly\extensions\trellis2\generator.py", line 680, in generate
return self._run_generate(image_bytes, params, progress_cb, cancel_event)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\Users\MSI\Documents\Modly\extensions\trellis2\generator.py", line 724, in _run_generate
mesh_with_voxel = self._model.run(
^^^^^^^^^^^^^^^^
File "C:\Users\MSI\Documents\Modly\extensions\trellis2\venv\Lib\site-packages\torch\utils_contextlib.py", line 116, in decorate_context
return func(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^
File "C:\Users\MSI\Documents\Modly\extensions\trellis2\venv\Lib\site-packages\trellis2_gguf\pipelines\trellis2_image_to_3d.py", line 1369, in run
out_mesh = self.decode_latent(shape_slat, None, res, use_tiled=use_tiled)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\Users\MSI\Documents\Modly\extensions\trellis2\venv\Lib\site-packages\torch\utils_contextlib.py", line 116, in decorate_context
return func(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^
File "C:\Users\MSI\Documents\Modly\extensions\trellis2\venv\Lib\site-packages\trellis2_gguf\pipelines\trellis2_image_to_3d.py", line 1024, in decode_latent
meshes, subs = self.decode_shape_slat(shape_slat, resolution, use_tiled=use_tiled)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\Users\MSI\Documents\Modly\extensions\trellis2\venv\Lib\site-packages\trellis2_gguf\pipelines\trellis2_image_to_3d.py", line 886, in decode_shape_slat
ret = self.models['shape_slat_decoder'](slat, return_subs=True, useTiled=use_tiled)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\Users\MSI\Documents\Modly\extensions\trellis2\venv\Lib\site-packages\torch\nn\modules\module.py", line 1751, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\Users\MSI\Documents\Modly\extensions\trellis2\venv\Lib\site-packages\torch\nn\modules\module.py", line 1762, in call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\Users\MSI\Documents\Modly\extensions\trellis2\venv\Lib\site-packages\trellis2_gguf\models\sc_vaes\fdg_vae.py", line 126, in forward
mesh = [Mesh(*tiled_flexible_dual_grid_to_mesh(
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\Users\MSI\Documents\Modly\extensions\trellis2\venv\Lib\site-packages\trellis2_gguf\models\sc_vaes\fdg_vae.py", line 126, in
mesh = [Mesh(*tiled_flexible_dual_grid_to_mesh(
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\Users\MSI\Documents\Modly\extensions\trellis2\venv\Lib\site-packages\torch\utils_contextlib.py", line 116, in decorate_context
return func(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^
File "C:\Users\MSI\Documents\Modly\extensions\trellis2\venv\Lib\site-packages\o_voxel\convert_init.py", line 25, in tiled_flexible_dual_grid_to_mesh
pbar = tqdm(total=len(x_range)*len(y_range)*len(z_range), desc="Fast Tiled Decoding")
^^^^
NameError: name 'tqdm' is not defined