blepping aad095ab14 Refactor to avoid having to monkeypatch forward_timestep_embed ...

Other cleanups

Allow setting a downscale factor and mode for CA downsampling

Add more type annotations

Make it so MSW-MSA attention failing is a warning rather than hard error

Adjust a few Ruff lint settings

2024-08-10 06:27:59 -06:00

..

__init__.py

Refactor to avoid having to monkeypatch forward_timestep_embed

2024-08-10 06:27:59 -06:00

msw_msa_attention.py

Refactor to avoid having to monkeypatch forward_timestep_embed

2024-08-10 06:27:59 -06:00

raunet.py

Refactor to avoid having to monkeypatch forward_timestep_embed

2024-08-10 06:27:59 -06:00

utils.py

Refactor to avoid having to monkeypatch forward_timestep_embed

2024-08-10 06:27:59 -06:00