fixed importing issue

Author: Serenagu525

src/maxdiffusion/models/ltx_video/transformers_pytorch/__init__.py

@@ -0,0 +1,16 @@
+# Copyright 2025 Lightricks Ltd.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://github.com/Lightricks/LTX-Video/blob/main/LICENSE
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+# This implementation is based on the Torch version available at:
+# https://github.com/Lightricks/LTX-Video/tree/main

src/maxdiffusion/models/ltx_video/transformers_pytorch/embeddings.py

@@ -1,3 +1,19 @@
+# Copyright 2025 Lightricks Ltd.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://github.com/Lightricks/LTX-Video/blob/main/LICENSE
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+# This implementation is based on the Torch version available at:
+# https://github.com/Lightricks/LTX-Video/tree/main
 # Adapted from: https://github.com/huggingface/diffusers/blob/main/src/diffusers/models/embeddings.py
 import math
 

src/maxdiffusion/models/ltx_video/transformers_pytorch/symmetric_patchifier.py

@@ -1,3 +1,19 @@
+# Copyright 2025 Lightricks Ltd.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://github.com/Lightricks/LTX-Video/blob/main/LICENSE
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+# This implementation is based on the Torch version available at:
+# https://github.com/Lightricks/LTX-Video/tree/main
 from abc import ABC, abstractmethod
 from typing import Tuple
 

src/maxdiffusion/models/ltx_video/transformers_pytorch/transformer3d.py

@@ -1,3 +1,19 @@
+# Copyright 2025 Lightricks Ltd.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://github.com/Lightricks/LTX-Video/blob/main/LICENSE
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+# This implementation is based on the Torch version available at:
+# https://github.com/Lightricks/LTX-Video/tree/main
 # Adapted from: https://github.com/huggingface/diffusers/blob/v0.26.3/src/diffusers/models/transformers/transformer_2d.py
 import math
 from dataclasses import dataclass

src/maxdiffusion/models/ltx_video/utils/convert_torch_weights_to_jax.py

@@ -218,14 +218,6 @@ def main(args):
         "the model parameters and optimizer state. This can affect optimizer moments and may result in a spike in "
         "training loss when resuming from the converted checkpoint."
     )
-  # print("Downloading files from GitHub...")
-  # github_url = "https://raw.githubusercontent.com/Lightricks/LTX-Video/main/ltx_video/models/transformers/"
-  # ltx_repo_path = "../"
-  # target_folder = "transformers_pytorch"
-  # files = ["attention.py", "embeddings.py", "symmetric_patchifier.py", "transformer3d.py"]
-  # module_path = "maxdiffusion.models.ltx_video.transformers_pytorch.transformer3d"
-
-  # Transformer3DModel = download_and_move_files(github_url, ltx_repo_path, target_folder, files, module_path)
 
   print("Loading safetensors, flush = True")
   weight_file = "ltxv-13b-0.9.7-dev.safetensors"