kijai 25eeab3c4c torch.compile support ...

works in Windows with torch 2.5.0 and Triton from https://github.com/woct0rdho/triton-windows

2024-10-25 18:15:30 +03:00

..

__pycache__

cleanup

2024-10-23 15:45:20 +03:00

__init__.py

initial commit

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

asymm_models_joint.py

torch.compile support

2024-10-25 18:15:30 +03:00

context_parallel.py

initial commit

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

layers.py

initial commit

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

mod_rmsnorm.py

torch.compile support

2024-10-25 18:15:30 +03:00

residual_tanh_gated_rmsnorm.py

initial commit

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

rope_mixed.py

initial commit

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

temporal_rope.py

initial commit

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

utils.py

initial commit

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