• Sayak Paul's avatar
    [Text-to-video] Add `torch.compile()` compatibility (#3949) · b62d9a1f
    Sayak Paul authored
    * use sample directly instead of the dataclass.
    
    * more usage of directly samples instead of dataclasses
    
    * more usage of directly samples instead of dataclasses
    
    * use direct sample in the pipeline.
    
    * direct usage of sample in the img2img case.
    b62d9a1f
unet_3d_blocks.py 23.4 KB