Add inference example project.

Author: spectralflight

README.md

@@ -59,10 +59,10 @@ Cosmos-Reason1 is included in [`transformers>=4.51.3`](https://huggingface.co/do
 
 We provide example inference scripts:
 
-* [Minimal example](scripts/inference_sample.py)
+* [Minimal example](examples/inference/inference.py)
 
   ```shell
-  ./scripts/inference_sample.py
+  uv run --project examples/project examples/project/inference.py
   ```
 
 * [Full example](scripts/inference.py)

examples/inference/.python-version

@@ -0,0 +1 @@
+3.10

scripts/inference_sample.py examples/inference/inference.pyscripts/inference_sample.py renamed to examples/inference/inference.py

@@ -1,4 +1,3 @@
-#!/usr/bin/env -S uv run --script
 # SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 #
@@ -14,45 +13,14 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-# https://docs.astral.sh/uv/guides/scripts/#using-a-shebang-to-create-an-executable-file
-# /// script
-# requires-python = ">=3.10"
-# dependencies = [
-#   "accelerate>=1.10.1",
-#   "qwen-vl-utils>=0.0.11",
-#   "torchcodec>=0.6.0",
-#   "torch>=2.7.1",
-#   "transformers>=4.51.3",
-#   "vllm>=0.10.1.1",
-# ]
-# [tool.uv.sources]
-# torch = [
-#   { index = "pytorch-cu128"},
-# ]
-# torchvision = [
-#   { index = "pytorch-cu128"},
-# ]
-# [[tool.uv.index]]
-# name = "pytorch-cu128"
-# url = "https://download.pytorch.org/whl/cu128"
-# explicit = true
-# ///
-
-"""Minimal example of inference with Cosmos-Reason1.
-
-Example:
-
-```shell
-./scripts/inference_sample.py
-```
-"""
+"""Minimal example of inference with Cosmos-Reason1."""
 
 from pathlib import Path
 
 import qwen_vl_utils
 import transformers
 
-ROOT = Path(__file__).parents[1]
+ROOT = Path(__file__).parents[2]
 SEPARATOR = "-" * 20
 
 
@@ -114,4 +82,4 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
+    main()

examples/inference/pyproject.toml

@@ -0,0 +1,23 @@
+[project]
+name = "cosmos-reason1-inference"
+version = "0.1.0"
+requires-python = ">=3.10"
+dependencies = [
+  "accelerate>=1.10.1",
+  "qwen-vl-utils>=0.0.11",
+  "torch>=2.7.1",
+  "torchcodec>=0.6.0",
+  "transformers>=4.51.3",
+]
+[tool.uv.sources]
+torch = [
+  { index = "pytorch-cu128"},
+]
+torchvision = [
+  { index = "pytorch-cu128"},
+]
+
+[[tool.uv.index]]
+name = "pytorch-cu128"
+url = "https://download.pytorch.org/whl/cu128"
+explicit = true