vidaio: upload miner files

chute_config.yml

@@ -1,5 +1,5 @@
 Image:
-  from_base: parachutes/python:3.12
+  from_base: parachutes/python:3.12-cu13  #needs to align with `libnvidia-encode` install to initialise NVENC sessions for av1_nvenc and hevc_nvenc compression
   set_user: root
   run_command:
     - apt-get update && apt-get install -y build-essential ninja-build nasm clang wget unzip nvidia-cuda-dev nvidia-cuda-toolkit pkg-config libx264-dev libx265-dev libvpx-dev git python3-pip python3-venv
@@ -10,7 +10,8 @@ Image:
   run_command:
     - wget https://github.com/FFmpeg/nv-codec-headers/archive/876af32a202d0de83bd1d36fe74ee0f7fcf86b0d.zip && unzip 876af32a202d0de83bd1d36fe74ee0f7fcf86b0d.zip && cd nv-codec-headers-876af32a202d0de83bd1d36fe74ee0f7fcf86b0d && make && make install && cd / && rm -rf nv-codec-headers-* 876af32a202d0de83bd1d36fe74ee0f7fcf86b0d.zip
     - git clone --depth 1 --branch 1.4.2 https://code.videolan.org/videolan/dav1d.git && cd dav1d && mkdir build && cd build && meson setup .. --default-library=static --buildtype release && ninja && ninja install && ldconfig && cd / && rm -rf dav1d
-    - export PKG_CONFIG_PATH="/usr/local/lib/pkgconfig:/usr/local/lib/x86_64-linux-gnu/pkgconfig:$PKG_CONFIG_PATH" && git clone https://github.com/FFmpeg/FFmpeg ffmpeg && cd ffmpeg && git reset --hard 33b215d1554a14e87416a24f8e6034312e629af7 && ./configure --enable-gpl --enable-libx264 --enable-libx265 --enable-libvpx --enable-libdav1d --enable-nonfree --enable-cuda-nvcc --enable-libnpp --extra-cflags="-I/usr/local/cuda/include" --extra-ldflags="-L/usr/local/cuda/lib64" --disable-static --enable-shared --nvccflags="-gencode arch=compute_75,code=sm_75 -gencode arch=compute_75,code=compute_75 -O2" && make -j && make install && ldconfig && cd / && rm -rf ffmpeg
+    # Build ffmpeg (pinned commit 33b215d) with CUDA + x264 + dav1d, no VMAF, no libnpp
+    - export PKG_CONFIG_PATH="/usr/local/lib/pkgconfig:/usr/local/lib/x86_64-linux-gnu/pkgconfig:$PKG_CONFIG_PATH" && git init ffmpeg && cd ffmpeg && git remote add origin https://github.com/FFmpeg/FFmpeg.git && git fetch --depth 1 origin 33b215d1554a14e87416a24f8e6034312e629af7 && git checkout FETCH_HEAD && ./configure --enable-gpl --enable-libx264 --enable-libx265 --enable-libvpx --enable-libdav1d --enable-nonfree --enable-cuda-nvcc --extra-cflags="-I/usr/local/cuda/include" --extra-ldflags="-L/usr/local/cuda/lib64" --disable-static --enable-shared --nvccflags="-gencode arch=compute_75,code=sm_75 -gencode arch=compute_75,code=compute_75 -O2" && make -j$(nproc) && make install && ldconfig && cd / && rm -rf ffmpeg
   set_user: chutes
   run_command:
     - uv pip install --upgrade setuptools wheel

miner.py

@@ -30,6 +30,10 @@ class Miner:
     def __repr__(self) -> str:
         return "CompressionMiner(ffmpeg)"
 
+    @staticmethod
+    def _is_nvenc(encoder: str) -> bool:
+        return "nvenc" in encoder
+
     def _vmaf_to_crf(self, vmaf: float, encoder: str) -> int:
         if "x264" in encoder or "x265" in encoder:
             if vmaf >= 93:
@@ -38,6 +42,14 @@ class Miner:
                 crf = 23
             else:
                 crf = 28
+        elif self._is_nvenc(encoder):
+            # NVENC uses -cq values (0-51 scale, lower = better quality)
+            if vmaf >= 93:
+                crf = 19
+            elif vmaf >= 89:
+                crf = 24
+            else:
+                crf = 30
         else:
             if vmaf >= 93:
                 crf = 25
@@ -45,7 +57,7 @@ class Miner:
                 crf = 30
             else:
                 crf = 35
-        print(f"[miner] VMAF threshold {vmaf} with encoder {encoder} -> CRF {crf}")
+        print(f"[miner] VMAF threshold {vmaf} with encoder {encoder} -> CRF/CQ {crf}")
         return crf
 
     @staticmethod
@@ -89,8 +101,12 @@ class Miner:
 
         if codec_mode == "CRF":
             crf = self._vmaf_to_crf(vmaf_threshold, encoder)
-            cmd += ["-c:v", encoder, "-crf", str(crf)]
-            print(f"[miner] Using CRF mode: crf={crf}")
+            if self._is_nvenc(encoder):
+                cmd += ["-c:v", encoder, "-rc", "vbr", "-cq", str(crf)]
+                print(f"[miner] Using NVENC CQ mode: cq={crf}")
+            else:
+                cmd += ["-c:v", encoder, "-crf", str(crf)]
+                print(f"[miner] Using CRF mode: crf={crf}")
         elif codec_mode == "CBR":
             cmd += ["-c:v", encoder, "-b:v", f"{target_bitrate}M"]
             print(f"[miner] Using CBR mode: bitrate={target_bitrate}M")