search

jiuntian / interactdiffusion

[CVPR 2024] Official repo for "InteractDiffusion: Interaction-Control for Text-to-Image Diffusion Model".
https://jiuntian.github.io/interactdiffusion/
94 stars 9 forks source link

Diffusers codes: ModuleNotFoundError: No module named 'interactdiffusion_unet_2d_condition' #3

Closed Kenneth-Wong closed 5 months ago

Kenneth-Wong commented 6 months ago

When I run the diffuser code : No module named 'interactdiffusion_unet_2d_condition'.

Are sth. missed?

jiuntian commented 6 months ago

Please check the colab notebook for the usage: https://colab.research.google.com/drive/1Bh9PjfTylxI2rbME5mQJtFqNTGvaghJq?usp=sharing I have just re-ran the code and it is working correctly.

The diffusers code: https://huggingface.co/interactdiffusion/diffusers-v1-2

If you wish to clone to local and perform changes to the code, you may initialise the pipeline as the following:

pipeline = DiffusionPipeline.from_pretrained(
    "<local path to cloned interactdiffusion diffusers code>",
    variant="fp16", torch_dtype=torch.float16
)

The interactdiffusion_unet_2d_condition.py is in unet folder.

Feel free to leave the comment if the problem persists.