Yoshimasa Niwa 99285ca1e7 Use float16 for autocast on mps ...

mps only supports float16 for autocast for now.

2024-11-05 13:53:34 +09:00

..

dit/joint_model

Add different RMSNorm functions for testing

2024-11-04 14:11:58 +02:00

vae

Workaround pad problem on mps

2024-11-05 12:46:13 +09:00

__init__.py

initial commit

2024-10-23 15:34:22 +03:00

t2v_synth_mochi.py

Use float16 for autocast on mps

2024-11-05 13:53:34 +09:00

utils.py

initial commit

2024-10-23 15:34:22 +03:00