Update app.py

app.py

@@ -104,6 +104,32 @@ RESOLUTIONS = {
 class LTX23DistilledA2VPipeline:
     """DistilledPipeline with optional audio conditioning."""
 
+    def __init__(
+        self,
+        distilled_checkpoint_path: str,
+        gemma_root: str,
+        spatial_upsampler_path: str,
+        loras: tuple,
+        quantization: QuantizationPolicy | None = None,
+    ):
+        self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+        self.dtype = torch.bfloat16
+
+        self.model_ledger = ModelLedger(
+            dtype=self.dtype,
+            device=self.device,
+            checkpoint_path=distilled_checkpoint_path,
+            spatial_upsampler_path=spatial_upsampler_path,
+            gemma_root_path=gemma_root,
+            loras=loras,
+            quantization=quantization,
+        )
+
+        self.pipeline_components = PipelineComponents(
+            dtype=self.dtype,
+            device=self.device,
+        )
+    
     def __call__(
         self,
         prompt: str,