add try-catch to make sure the log file can be downloaded

helper.py

@@ -392,48 +392,53 @@ def generate_video_fun(checkpoints_path: str):
 
         log.info(f"actual_seed: {actual_seed}")
 
-        if rgb_video_path is None or not os.path.isfile(rgb_video_path):
-            log.warning(f"File `{rgb_video_path}` does not exist")
-            rgb_video_path = ""
-
-        # add timer to calculate the generation time
-        start_time = time.time()
-
-        # parse generation configs
-        args, control_inputs = parse_arguments(
-            controlnet_specs_in={
-                "hdmap": {"control_weight": 0.3, "input_control": hdmap_video_input},
-                "lidar": {"control_weight": 0.7, "input_control": lidar_video_input},
-            },
-            input_video_path=rgb_video_path,
-            checkpoint_dir=checkpoints_path,
-            prompt=prompt,
-            negative_prompt=negative_prompt,
-            sigma_max=80,
-            offload_text_encoder_model=True,
-            is_av_sample=True,
-            num_gpus=1,
-            seed=seed,
-        )
+        try:
+            if rgb_video_path is None or not os.path.isfile(rgb_video_path):
+                log.warning(f"File `{rgb_video_path}` does not exist")
+                rgb_video_path = ""
+
+            # add timer to calculate the generation time
+            start_time = time.time()
+
+            # parse generation configs
+            args, control_inputs = parse_arguments(
+                controlnet_specs_in={
+                    "hdmap": {"control_weight": 0.3, "input_control": hdmap_video_input},
+                    "lidar": {"control_weight": 0.7, "input_control": lidar_video_input},
+                },
+                input_video_path=rgb_video_path,
+                checkpoint_dir=checkpoints_path,
+                prompt=prompt,
+                negative_prompt=negative_prompt,
+                sigma_max=80,
+                offload_text_encoder_model=True,
+                is_av_sample=True,
+                num_gpus=1,
+                seed=seed,
+            )
 
-        # watch gpu memory
-        watcher = watch_gpu_memory(10, lambda x: log.debug(f"GPU memory (used, total): {x} (MiB)"))
+            # watch gpu memory
+            watcher = watch_gpu_memory(10, lambda x: log.debug(f"GPU memory (used, total): {x} (MiB)"))
 
-        # start inference
-        if chunking <= 0:
-            chunking = None
-        videos, prompts = inference(args, control_inputs, chunking)
+            # start inference
+            if chunking <= 0:
+                chunking = None
+            videos, prompts = inference(args, control_inputs, chunking)
 
-        # print the generation time
-        end_time = time.time()
-        log.info(f"Time taken: {end_time - start_time} s")
+            # print the generation time
+            end_time = time.time()
+            log.info(f"Time taken: {end_time - start_time} s")
 
-        # stop the watcher
-        stop_watcher()
+            # stop the watcher
+            stop_watcher()
 
-        video = videos[0]
+            video = videos[0]
 
-        log.logger.remove(log_handler)
-        return video, create_zip_for_download(filename=logfile_path, files_to_zip=[video, logfile_path]), actual_seed
+            log.logger.remove(log_handler)
+            return video, create_zip_for_download(filename=logfile_path, files_to_zip=[video, logfile_path]), actual_seed
+        except Exception as e:
+            log.logger.remove(log_handler)
+            log.exception(e)
+            return "", logfile_path, actual_seed
 
     return generate_video