gluttony-10 dd58129e07 Update pyramid_dit_for_video_gen_pipeline.py ...

Change "torch.float8_e4m3fn, torch.float8_e4m3fn" to "torch.float8_e4m3fn, torch.float8_e5m2".

2024-10-11 10:25:45 +08:00

..

__init__.py

init

2024-10-10 13:16:35 +03:00

modeling_embedding.py

init

2024-10-10 13:16:35 +03:00

modeling_mmdit_block.py

fix bf16

2024-10-10 21:43:12 +03:00

modeling_normalization.py

fp8

2024-10-10 23:12:11 +03:00

modeling_pyramid_mmdit.py

fp8_fast mode, cleanup

2024-10-11 00:47:08 +03:00

modeling_text_encoder.py

init

2024-10-10 13:16:35 +03:00

pyramid_dit_for_video_gen_pipeline.py

Update pyramid_dit_for_video_gen_pipeline.py

2024-10-11 10:25:45 +08:00