Update api/ltx/ltx_utils.py

api/ltx/ltx_utils.py

@@ -108,15 +108,15 @@ def build_ltx_pipeline_on_cpu(config: Dict) -> Tuple[LTXVideoPipeline, Optional[
         prompt_enhancer_llm_model=None, prompt_enhancer_llm_tokenizer=None,
     )
 
-    #latent_upsampler = None
-    #if config.get("spatial_upscaler_model_path"):
-    #    spatial_path = config["spatial_upscaler_model_path"]
-    #    latent_upsampler = create_latent_upsampler(spatial_path, device="cpu")
-    #    if precision == "bfloat16":
-    #        latent_upsampler.to(torch.bfloat16)
+    latent_upsampler = None
+    if config.get("spatial_upscaler_model_path"):
+        spatial_path = config["spatial_upscaler_model_path"]
+        latent_upsampler = create_latent_upsampler(spatial_path, device="cpu")
+        if precision == "bfloat16":
+            latent_upsampler.to(torch.bfloat16)
 
     logging.info(f"LTX pipeline built on CPU in {time.perf_counter() - t0:.2f}s")
-    return pipeline, _ #latent_upsampler
+    return pipeline, latent_upsampler
 
 
 # ==============================================================================