LTX Video (#123)

* rename files

* ltx finetuning

* update

* update

* improvements

* make style

* gradient clipping

* update

* fix distributed inference

* update

* update

finetrainers/state.py

@@ -0,0 +1,23 @@
+import torch
+from accelerate import Accelerator
+
+
+class State:
+    # Training state
+    seed: int = None
+    model_name: str = None
+    accelerator: Accelerator = None
+    weight_dtype: torch.dtype = None
+    train_epochs: int = None
+    train_steps: int = None
+    overwrote_max_train_steps: bool = False
+    num_trainable_parameters: int = 0
+    learning_rate: float = None
+    train_batch_size: int = None
+    generator: torch.Generator = None
+
+    # Hub state
+    repo_id: str = None
+
+    # Artifacts state
+    output_dir: str = None