microsoft / microsoft/TRELLIS

[Error]"Specify the reduction dim with the 'dim' argument" occurs only with the text-xlarge model (includes temporary workaround)

Open
#216 2 comments 2 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

Dominant language
Python
Stars
13.7k
Forks
1.3k
PR merge metrics
No merged PRs in 30d

Description

Image

When running example_text.py, I encounter the following error:
RuntimeError: max(): Expected reduction dim to be specified for input.numel() == 0. Specify the reduction dim with the 'dim' argument.

Even though this error appears, I can still run the code in the following way
Looking at the example code for the Text to 3D approach, we see:
pipeline = TrellisTextTo3DPipeline.from_pretrained("JeffreyXiang/TRELLIS-text-xlarge")

However, the above error only occurs with the text-xlarge model.
As a workaround, if I download the text-large or text-base models locally and then run:
TrellisTextTo3DPipeline.from_pretrained("/path/to/TRELLIS-text-large")
or
TrellisTextTo3DPipeline.from_pretrained("/path/to/TRELLIS-text-base")

the code works fine. I suspect that the text-xlarge model might be incompatible with certain environments.
Below is the result of running example_text.py with the text-large model.
This workaround is working well for me, so you might consider it as an option.

Image

I'd appreciate it if you could provide a solution for resolving this error occurring with the text-xlarge model.

Contributor guide

No contributing guide indexed for this repository

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Start by reproducing the failure in example_text.py with TrellisTextTo3DPipeline.from_pretrained("JeffreyXiang/TRELLIS-text-xlarge"), then compare it with the text-large and text-base local models. Investigate the environment and model-loading differences visible from that entry point; done means the text-xlarge example runs without the reduction-dimension error.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
machine-learning
Issue type
Bug
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
32/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.