Skip to content
This repository was archived by the owner on Feb 7, 2025. It is now read-only.
This repository was archived by the owner on Feb 7, 2025. It is now read-only.

Fix Torchscript error in latent diffusion models unet network #31

Description

@Warvito

Try to include torchscript tests for the latent diffusion unet networks. For this, it might be necessary to remove the use of TimestepEmbedSequential and TimestepBlock by creating blocks exclusive to when we have the time embedding included.
A similar solution to Huggingface diffuser might be useful:
https://github.com/huggingface/diffusers/blob/2d35f6733a2d698e8917896071444a5923993ae7/src/diffusers/models/unet_blocks.py#L461
https://github.com/huggingface/diffusers/blob/2d35f6733a2d698e8917896071444a5923993ae7/src/diffusers/models/unet_blocks.py#L379
https://github.com/huggingface/diffusers/blob/2d35f6733a2d698e8917896071444a5923993ae7/src/diffusers/models/unet_blocks.py#L576

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type
No fields configured for issues without a type.

Projects

No projects

Relationships

None yet

Development

No branches or pull requests

Issue actions