Update final_video.py

Fixed for apple silicon machine. Now video successfully builds.
6 months ago · fe7f26686c
parent 0f24c8b7db
commit fe7f26686c
1 changed files with 3 additions and 3 deletions
--- a/video_creation/final_video.py
+++ b/video_creation/final_video.py
@ -93,7 +93,7 @@ def prepare_background(reddit_id: str, W: int, H: int) -> str:
            output_path,
            an=None,
            **{
-                "c:v": "h264_nvenc",
+                "c:v": "h264_nvenc" if os.uname().machine != 'arm64' else 'h264',
                "b:v": "20M",
                "b:a": "192k",
                "threads": multiprocessing.cpu_count(),
@ -438,7 +438,7 @@ def make_final_video(
                path,
                f="mp4",
                **{
-                    "c:v": "h264_nvenc",
+                    "c:v": "h264_nvenc" if os.uname().machine != 'arm64' else 'h264',
                    "b:v": "20M",
                    "b:a": "192k",
                    "threads": multiprocessing.cpu_count(),
@ -468,7 +468,7 @@ def make_final_video(
                    path,
                    f="mp4",
                    **{
-                        "c:v": "h264_nvenc",
+                        "c:v": "h264_nvenc" if os.uname().machine != 'arm64' else 'h264',
                        "b:v": "20M",
                        "b:a": "192k",
                        "threads": multiprocessing.cpu_count(),