-
Notifications
You must be signed in to change notification settings - Fork 487
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Program Hang/Stuck after using F.interpolate? VAE Decode Step of HunyuanVideo model. #8470
Comments
*Update: After a very long time, for 256x160x49 the program is running its next iteration. But I will still keep this issue up as it's taking a very long time to run. FULL LOGS:
|
|
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
🐛 Bug
I'm trying out the HunyuanVideo model on TPUs and it seems like everything works fine after changing where needed to work with pytorch xla. But when doing VAE decode at the end of the program, it is stuck after doing F.interpolate
To Reproduce
Steps to reproduce the behavior:
Both generation at 256x160x49 and 544x960x49 were tested, and the program hangs.
line that uses F.interpolate:
https://github.com/Tencent/HunyuanVideo/blob/c4a9d7708dac7c930181c9e147d0092dffa36f92/hyvideo/vae/unet_causal_3d_blocks.py#L159
LOGS:
Expected behavior
Environment
pip3 install torch==2.6.0.dev20241105+cpu torchvision torchaudio==2.5.0.dev20241105+cpu --index-url https://download.pytorch.org/whl/nightly/cpu
pip3 install https://storage.googleapis.com/pytorch-xla-releases/wheels/tpuvm/torch_xla-2.6.0.dev20241105-cp310-cp310-linux_x86_64.whl
Additional context
The text was updated successfully, but these errors were encountered: