File tree Expand file tree Collapse file tree
src/maxdiffusion/models/ltx_video/transformers Expand file tree Collapse file tree Original file line number Diff line number Diff line change 1+ # Copyright 2025 Lightricks Ltd.
2+ #
3+ # Licensed under the Apache License, Version 2.0 (the "License");
4+ # you may not use this file except in compliance with the License.
5+ # You may obtain a copy of the License at
6+ #
7+ # https://github.com/Lightricks/LTX-Video/blob/main/LICENSE
8+ #
9+ # Unless required by applicable law or agreed to in writing, software
10+ # distributed under the License is distributed on an "AS IS" BASIS,
11+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12+ # See the License for the specific language governing permissions and
13+ # limitations under the License.
14+ #
15+ # This implementation is based on the Torch version available at:
16+ # https://github.com/Lightricks/LTX-Video/tree/main
117from typing import Optional , Tuple
218
319import jax
Original file line number Diff line number Diff line change 1+ # Copyright 2025 Lightricks Ltd.
2+ #
3+ # Licensed under the Apache License, Version 2.0 (the "License");
4+ # you may not use this file except in compliance with the License.
5+ # You may obtain a copy of the License at
6+ #
7+ # https://github.com/Lightricks/LTX-Video/blob/main/LICENSE
8+ #
9+ # Unless required by applicable law or agreed to in writing, software
10+ # distributed under the License is distributed on an "AS IS" BASIS,
11+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12+ # See the License for the specific language governing permissions and
13+ # limitations under the License.
14+ #
15+ # This implementation is based on the Torch version available at:
16+ # https://github.com/Lightricks/LTX-Video/tree/main
117from typing import Dict , Optional , Tuple
218
319import jax
Original file line number Diff line number Diff line change 1+ # Copyright 2025 Lightricks Ltd.
2+ #
3+ # Licensed under the Apache License, Version 2.0 (the "License");
4+ # you may not use this file except in compliance with the License.
5+ # You may obtain a copy of the License at
6+ #
7+ # https://github.com/Lightricks/LTX-Video/blob/main/LICENSE
8+ #
9+ # Unless required by applicable law or agreed to in writing, software
10+ # distributed under the License is distributed on an "AS IS" BASIS,
11+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12+ # See the License for the specific language governing permissions and
13+ # limitations under the License.
14+ #
15+ # This implementation is based on the Torch version available at:
16+ # https://github.com/Lightricks/LTX-Video/tree/main
117from functools import partial
218import math
319from typing import Any , Dict , Optional , Tuple
Original file line number Diff line number Diff line change 1+ # Copyright 2025 Lightricks Ltd.
2+ #
3+ # Licensed under the Apache License, Version 2.0 (the "License");
4+ # you may not use this file except in compliance with the License.
5+ # You may obtain a copy of the License at
6+ #
7+ # https://github.com/Lightricks/LTX-Video/blob/main/LICENSE
8+ #
9+ # Unless required by applicable law or agreed to in writing, software
10+ # distributed under the License is distributed on an "AS IS" BASIS,
11+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12+ # See the License for the specific language governing permissions and
13+ # limitations under the License.
14+ #
15+ # This implementation is based on the Torch version available at:
16+ # https://github.com/Lightricks/LTX-Video/tree/main
117from flax import linen as nn
218import jax .numpy as jnp
319
Original file line number Diff line number Diff line change 1+ # Copyright 2025 Lightricks Ltd.
2+ #
3+ # Licensed under the Apache License, Version 2.0 (the "License");
4+ # you may not use this file except in compliance with the License.
5+ # You may obtain a copy of the License at
6+ #
7+ # https://github.com/Lightricks/LTX-Video/blob/main/LICENSE
8+ #
9+ # Unless required by applicable law or agreed to in writing, software
10+ # distributed under the License is distributed on an "AS IS" BASIS,
11+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12+ # See the License for the specific language governing permissions and
13+ # limitations under the License.
14+ #
15+ # This implementation is based on the Torch version available at:
16+ # https://github.com/Lightricks/LTX-Video/tree/main
117from typing import List , Optional , Tuple
218
319import jax
You can’t perform that action at this time.
0 commit comments