Fix arm inference sample.

Author: spectralflight

cosmos_reason2_utils/pyproject.toml

@@ -32,7 +32,7 @@ dependencies = [
   "pillow>=12.0.0",
   "pydantic>=2.12.4",
   "pyyaml>=6.0.3",
-  "qwen-vl-utils>=0.0.14",
+  "qwen-vl-utils==0.0.14",
   "rich>=14.2.0",
   "transformers>=4.57.0,<5.0.0",
   "tyro>=0.9.35",
@@ -44,7 +44,7 @@ dev = [
     "ruff==0.14.8",
 ]
 cosmos-rl = [
-  "cosmos-rl>=0.3.6",
+  "cosmos-rl==0.3.6",
   "datasets>=4.4.1",
   "toml>=0.10.2",
   "wandb>=0.23.0",

examples/cosmos_rl/uv.lock

@@ -18,11 +18,12 @@
 # requires-python = ">=3.10"
 # dependencies = [
 #   "accelerate==1.12.0",
+#   "av==16.1.0",
 #   "pillow==12.0.0",
 #   "transformers==4.57.3",
 #   "torch==2.9.0",
 #   "torchvision",
-#   "torchcodec==0.9.1",
+#   "torchcodec==0.9.1; platform_machine != 'aarch64'",
 # ]
 # ///