Skip to content

Using AutoModel.from_pretrained for wan2.1 crash my notebook #11996

@Javierxx05

Description

@Javierxx05

Hi, i been trying to use the diffusers library to use the Wan2.1 model, and i was following the official documentation to use it with the diffusers library from here: https://huggingface.co/docs/diffusers/main/en/api/pipelines/wan

But everytime I try to load the vae my jupyter notebook, It keep crashing when the model its loaded at 50%. whith the next message:
vae = AutoModel.from_pretrained("Wan-AI/Wan2.1-T2V-14B-Diffusers", subfolder="vae", torch_dtype=torch.float32)

The Kernel crashed while executing code in the current cell or a previous cell. Please review the code in the cell(s) to identify a possible cause of the failure. Click [here](https://aka.ms/vscodeJupyterKernelCrash) for more info. View Jupyter [log](command:jupyter.viewOutput) for further details.

The log doesn't show any more info that indicates the error.

My computer info:
Windows 11 in WSL
conda with python 3.11
RTX5090
RAM 64gb
Intel core i7-12700

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions