Generation failed
Error: Traceback (most recent call last):
File "C:\Modly\resources\api\routers\generation.py", line 170, in _run_generation
output_path = await loop.run_in_executor(
^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\Modly\resources\python-embed\Lib\concurrent\futures\thread.py", line 58, in run
result = self.fn(*self.args, **self.kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\Modly\resources\api\routers\generation.py", line 172, in
lambda: gen.generate(image_bytes, params, progress_cb, cancel_event)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\Modly\resources\api\services\extension_process.py", line 240, in generate
raise RuntimeError(msg.get("traceback") or msg.get("message", "Unknown error"))
RuntimeError: Traceback (most recent call last):
File "C:\Modly\resources\api\runner.py", line 160, in main
output_path = gen.generate(image_bytes, params, progress_cb, cancel_evt)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "D:\Modly\extensions\trellis2\generator.py", line 680, in generate
return self._run_generate(image_bytes, params, progress_cb, cancel_event)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "D:\Modly\extensions\trellis2\generator.py", line 724, in _run_generate
mesh_with_voxel = self.model.run(
^^^^^^^^^^^^^^^^
File "D:\Modly\extensions\trellis2\venv\Lib\site-packages\torch\utils_contextlib.py", line 116, in decorate_context
return func(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^
File "D:\Modly\extensions\trellis2\venv\Lib\site-packages\trellis2_gguf\pipelines\trellis2_image_to_3d.py", line 1593, in run
self.load_sparse_structure_model()
File "D:\Modly\extensions\trellis2\venv\Lib\site-packages\trellis2_gguf\pipelines\trellis2_image_to_3d.py", line 694, in load_sparse_structure_model
self.models['sparse_structure_flow_model'] = models.from_pretrained(
^^^^^^^^^^^^^^^^^^^^^^^
File "D:\Modly\extensions\trellis2\venv\Lib\site-packages\trellis2_gguf\models_init.py", line 353, in from_pretrained
model.load_state_dict(sd, strict=False)
File "D:\Modly\extensions\trellis2\venv\Lib\site-packages\torch\nn\modules\module.py", line 2581, in load_state_dict
raise RuntimeError(
RuntimeError: Error(s) in loading state_dict for SparseStructureFlowModel:
While copying the parameter named "blocks.0.self_attn.to_qkv.weight", whose dimensions in the model are torch.Size([4608, 1536]) and whose dimensions in the checkpoint are torch.Size([4608, 1536]), an exception occurred : ('The size of tensor a (1536) must match the size of tensor b (1632) at non-singleton dimension 1',).
While copying the parameter named "blocks.0.self_attn.to_out.weight", whose dimensions in the model are torch.Size([1536, 1536]) and whose dimensions in the checkpoint are torch.Size([1536, 1536]), an exception occurred : ('The size of tensor a (1536) must match the size of tensor b (1632) at non-singleton dimension 1',).
While copying the parameter named "blocks.0.cross_attn.to_q.weight", whose dimensions in the model are torch.Size([1536, 1536]) and whose dimensions in the checkpoint are torch.Size([1536, 1536]), an exception occurred : ('The size of tensor a (1536) must match the size of tensor b (1632) at non-singleton dimension 1',).
While copying the parameter named "blocks.0.cross_attn.to_kv.weight", whose dimensions in the model are torch.Size([3072, 1024]) and whose dimensions in the checkpoint are torch.Size([3072, 1024]), an exception occurred : ('The size of tensor a (1024) must match the size of tensor b (1088) at non-singleton dimension 1',).
While copying the parameter named "blocks.0.cross_attn.to_out.weight", whose dimensions in the model are torch.Size([1536, 1536]) and whose dimensions in the checkpoint are torch.Size([1536, 1536]), an exception occurred : ('The size of tensor a (1536) must match the size of tensor b (1632) at non-singleton dimension 1',).
While copying the parameter named "blocks.0.mlp.mlp.0.weight", whose dimensions in the model are torch.Size([8192, 1536]) and whose dimensions in the checkpoint are torch.Size([8192, 1536]), an exception occurred : ('The size of tensor a (1536) must match the size of tensor b (1632) at non-singleton dimension 1',).
While copying the parameter named "blocks.0.mlp.mlp.2.weight", whose dimensions in the model are torch.Size([1536, 8192]) and whose dimensions in the checkpoint are torch.Size([1536, 8192]), an exception occurred : ('The size of tensor a (8192) must match the size of tensor b (8704) at non-singleton dimension 1',).
While copying the parameter named "blocks.1.self_attn.to_qkv.weight", whose dimensions in the model are torch.Size([4608, 1536]) and whose dimensions in the checkpoint are torch.Size([4608, 1536]), an exception occurred : ('The size of tensor a (1536) must match the size of tensor b (1632) at non-singleton dimension 1',).
While copying the parameter named "blocks.1.self_attn.to_out.weight", whose dimensions in the model are torch.Size([1536, 1536]) and whose dimensions in the checkpoint are torch.Size([1536, 1536]), an exception occurred : ('The size of tensor a (1536) must match the size of tensor b (1632) at non-singleton dimension 1',).
While copying the parameter named "blocks.1.cross_attn.to_q.weight", whose dimensions in the model are torch.Size([1536, 1536]) and whose dimensions in the checkpoint are torch.Size([1536, 1536]), an exception occurred : ('The size of tensor a (1536) must match the size of tensor b (1632) at non-singleton dimension 1',).
While copying the parameter named "blocks.1.cross_attn.to_kv.weight", whose dimensions in the model are torch.Size([3072, 1024]) and whose dimensions in the checkpoint are torch.Size([3072, 1024]), an exception occurred : ('The size of tensor a (1024) must match the size of tensor b (1088) at non-singleton dimension 1',).
While copying the parameter named "blocks.1.cross_attn.to_out.weight", whose dimensions in the model are torch.Size([1536, 1536]) and whose dimensions in the checkpoint are torch.Size([1536, 1536]), an exception occurred : ('The size of tensor a (1536) must match the size of tensor b (1632) at non-singleton dimension 1',).
While copying the parameter named "blocks.1.mlp.mlp.0.weight", whose dimensions in the model are torch.Size([8192, 1536]) and whose dimensions in the checkpoint are torch.Size([8192, 1536]), an exception occurred : ('The size of tensor a (1536) must match the size of tensor b (1632) at non-singleton dimension 1',).
While copying the parameter named "blocks.1.mlp.mlp.2.weight", whose dimensions in the model are torch.Size([1536, 8192]) and whose dimensions in the checkpoint are torch.Size([1536, 8192]), an exception occurred : ('The size of tensor a (8192) must match the size of tensor b (8704) at non-singleton dimension 1',).
While copying the parameter named "blocks.2.self_attn.to_qkv.weight", whose dimensions in the model are torch.Size([4608, 1536]) and whose dimensions in the checkpoint are torch.Size([4608, 1536]), an exception occurred : ('The size of tensor a (1536) must match the size of tensor b (1632) at non-singleton dimension 1',).
While copying the parameter named "blocks.2.self_attn.to_out.weight", whose dimensions in the model are torch.Size([1536, 1536]) and whose dimensions in the checkpoint are torch.Size([1536, 1536]), an exception occurred : ('The size of tensor a (1536) must match the size of tensor b (1632) at non-singleton dimension 1',).
While copying the parameter named "blocks.2.cross_attn.to_q.weight", whose dimensions in the model are torch.Size([1536, 1536]) and whose dimensions in the checkpoint are torch.Size([1536, 1536]), an exception occurred : ('The size of tensor a (1536) must match the size of tensor b (1632) at non-singleton dimension 1',).
While copying the parameter named "blocks.2.cross_attn.to_kv.weight", whose dimensions in the model are torch.Size([3072, 1024]) and whose dimensions in the checkpoint are torch.Size([3072, 1024]), an exception occurred : ('The size of tensor a (1024) must match the size of tensor b (1088) at non-singleton dimension 1',).
While copying the parameter named "blocks.2.cross_attn.to_out.weight", whose dimensions in the model are torch.Size([1536, 1536]) and whose dimensions in the checkpoint are torch.Size([1536, 1536]), an exception occurred : ('The size of tensor a (1536) must match the size of tensor b (1632) at non-singleton dimension 1',).
While copying the parameter named "blocks.2.mlp.mlp.0.weight", whose dimensions in the model are torch.Size([8192, 1536]) and whose dimensions in the checkpoint are torch.Size([8192, 1536]), an exception occurred : ('The size of tensor a (1536) must match the size of tensor b (1632) at non-singleton dimension 1',).
While copying the parameter named "blocks.2.mlp.mlp.2.weight", whose dimensions in the model are torch.Size([1536, 8192]) and whose dimensions in the checkpoint are torch.Size([1536, 8192]), an exception occurred : ('The size of tensor a (8192) must match the size of tensor b (8704) at non-singleton dimension 1',).
While copying the parameter named "blocks.3.self_attn.to_qkv.weight", whose dimensions in the model are torch.Size([4608, 1536]) and whose dimensions in the checkpoint are torch.Size([4608, 1536]), an exception occurred : ('The size of tensor a (1536) must match the size of tensor b (1632) at non-singleton dimension 1',).
While copying the parameter named "blocks.3.self_attn.to_out.weight", whose dimensions in the model are torch.Size([1536, 1536]) and whose dimensions in the checkpoint are torch.Size([1536, 1536]), an exception occurred : ('The size of tensor a (1536) must match the size of tensor b (1632) at non-singleton dimension 1',).
While copying the parameter named "blocks.3.cross_attn.to_q.weight", whose dimensions in the model are torch.Size([1536, 1536]) and whose dimensions in the checkpoint are torch.Size([1536, 1536]), an exception occurred : ('The size of tensor a (1536) must match the size of tensor b (1632) at non-singleton dimension 1',).
While copying the parameter named "blocks.3.cross_attn.to_kv.weight", whose dimensions in the model are torch.Size([3072, 1024]) and whose dimensions in the checkpoint are torch.Size([3072, 1024]), an exception occurred : ('The size of tensor a (1024) must match the size of tensor b (1088) at non-singleton dimension 1',).
While copying the parameter named "blocks.3.cross_attn.to_out.weight", whose dimensions in the model are torch.Size([1536, 1536]) and whose dimensions in the checkpoint are torch.Size([1536, 1536]), an exception occurred : ('The size of tensor a (1536) must match the size of tensor b (1632) at non-singleton dimension 1',).
While copying the parameter named "blocks.3.mlp.mlp.0.weight", whose dimensions in the model are torch.Size([8192, 1536]) and whose dimensions in the checkpoint are torch.Size([8192, 1536]), an exception occurred : ('The size of tensor a (1536) must match the size of tensor b (1632) at non-singleton dimension 1',).
While copying the parameter named "blocks.3.mlp.mlp.2.weight", whose dimensions in the model are torch.Size([1536, 8192]) and whose dimensions in the checkpoint are torch.Size([1536, 8192]), an exception occurred : ('The size of tensor a (8192) must match the size of tensor b (8704) at non-singleton dimension 1',).
While copying the parameter named "blocks.4.self_attn.to_qkv.weight", whose dimensions in the model are torch.Size([4608, 1536]) and whose dimensions in the checkpoint are torch.Size([4608, 1536]), an exception occurred : ('The size of tensor a (1536) must match the size of tensor b (1632) at non-singleton dimension 1',).
While copying the parameter named "blocks.4.self_attn.to_out.weight", whose dimensions in the model are torch.Size([1536, 1536]) and whose dimensions in the checkpoint are torch.Size([1536, 1536]), an exception occurred : ('The size of tensor a (1536) must match the size of tensor b (1632) at non-singleton dimension 1',).
While copying the parameter named "blocks.4.cross_attn.to_q.weight", whose dimensions in the model are torch.Size([1536, 1536]) and whose dimensions in the checkpoint are torch.Size([1536, 1536]), an exception occurred : ('The size of tensor a (1536) must match the size of tensor b (1632) at non-singleton dimension 1',).
While copying the parameter named "blocks.4.cross_attn.to_kv.weight", whose dimensions in the model are torch.Size([3072, 1024]) and whose dimensions in the checkpoint are torch.Size([3072, 1024]), an exception occurred : ('The size of tensor a (1024) must match the size of tensor b (1088) at non-singleton dimension 1',).
While copying the parameter named "blocks.4.cross_attn.to_out.weight", whose dimensions in the model are torch.Size([1536, 1536]) and whose dimensions in the checkpoint are torch.Size([1536, 1536]), an exception occurred : ('The size of tensor a (1536) must match the size of tensor b (1632) at non-singleton dimension 1',).
While copying the parameter named "blocks.4.mlp.mlp.0.weight", whose dimensions in the model are torch.Size([8192, 1536]) and whose dimensions in the checkpoint are torch.Size([8192, 1536]), an exception occurred : ('The size of tensor a (1536) must match the size of tensor b (1632) at non-singleton dimension 1',).
While copying the parameter named "blocks.4.mlp.mlp.2.weight", whose dimensions in the model are torch.Size([1536, 8192]) and whose dimensions in the checkpoint are torch.Size([1536, 8192]), an exception occurred : ('The size of tensor a (8192) must match the size of tensor b (8704) at non-singleton dimension 1',).
...
While copying the parameter named "blocks.28.mlp.mlp.2.weight", whose dimensions in the model are torch.Size([1536, 8192]) and whose dimensions in the checkpoint are torch.Size([1536, 8192]), an exception occurred : ('The size of tensor a (8192) must match the size of tensor b (8704) at non-singleton dimension 1',).
While copying the parameter named "blocks.29.self_attn.to_qkv.weight", whose dimensions in the model are torch.Size([4608, 1536]) and whose dimensions in the checkpoint are torch.Size([4608, 1536]), an exception occurred : ('The size of tensor a (1536) must match the size of tensor b (1632) at non-singleton dimension 1',).
While copying the parameter named "blocks.29.self_attn.to_out.weight", whose dimensions in the model are torch.Size([1536, 1536]) and whose dimensions in the checkpoint are torch.Size([1536, 1536]), an exception occurred : ('The size of tensor a (1536) must match the size of tensor b (1632) at non-singleton dimension 1',).
While copying the parameter named "blocks.29.cross_attn.to_q.weight", whose dimensions in the model are torch.Size([1536, 1536]) and whose dimensions in the checkpoint are torch.Size([1536, 1536]), an exception occurred : ('The size of tensor a (1536) must match the size of tensor b (1632) at non-singleton dimension 1',).
While copying the parameter named "blocks.29.cross_attn.to_kv.weight", whose dimensions in the model are torch.Size([3072, 1024]) and whose dimensions in the checkpoint are torch.Size([3072, 1024]), an exception occurred : ('The size of tensor a (1024) must match the size of tensor b (1088) at non-singleton dimension 1',).
While copying the parameter named "blocks.29.cross_attn.to_out.weight", whose dimensions in the model are torch.Size([1536, 1536]) and whose dimensions in the checkpoint are torch.Size([1536, 1536]), an exception occurred : ('The size of tensor a (1536) must match the size of tensor b (1632) at non-singleton dimension 1',).
While copying the parameter named "blocks.29.mlp.mlp.0.weight", whose dimensions in the model are torch.Size([8192, 1536]) and whose dimensions in the checkpoint are torch.Size([8192, 1536]), an exception occurred : ('The size of tensor a (1536) must match the size of tensor b (1632) at non-singleton dimension 1',).
While copying the parameter named "blocks.29.mlp.mlp.2.weight", whose dimensions in the model are torch.Size([1536, 8192]) and whose dimensions in the checkpoint are torch.Size([1536, 8192]), an exception occurred : ('The size of tensor a (8192) must match the size of tensor b (8704) at non-singleton dimension 1',).