kijai ea1b92cfc2 RoPE optimizations and fix ...

- small overall speed boost and fix freqs dtype
- allow not upcasting RoPE for slight speed boost

2025-01-17 16:57:37 +02:00

..

__init__.py

fix

2025-01-06 06:46:53 +02:00

activation_layers.py

initial commit

2024-12-03 19:09:50 +02:00

attention.py

initial STG testing

2024-12-06 17:09:56 +02:00

embed_layers.py

initial commit

2024-12-03 19:09:50 +02:00

fp8_map.safetensors

Support official scaled fp8 model

2024-12-18 10:29:56 +02:00

fp8_optimization.py

consistently use safe_load

2024-12-19 01:53:00 +02:00

mlp_layers.py

initial commit

2024-12-03 19:09:50 +02:00

models.py

RoPE optimizations and fix

2025-01-17 16:57:37 +02:00

modulate_layers.py

initial commit

2024-12-03 19:09:50 +02:00

norm_layers.py

initial commit

2024-12-03 19:09:50 +02:00

posemb_layers.py

RoPE optimizations and fix

2025-01-17 16:57:37 +02:00

token_refiner.py

Add alternative cpu offloading

2024-12-23 17:47:09 +02:00