From 4d9710fb2ee8ccaf6d8581c5183b635bfb2d8bd2 Mon Sep 17 00:00:00 2001
From: jianyuan <jianyuan@meta.com>
Date: Sun, 23 Mar 2025 20:07:49 +0000
Subject: [PATCH] update pyproject and readme

---
 README.md             | 11 +++++++++--
 pyproject.toml        | 19 ++++++++++---------
 requirements.txt      |  7 +++++++
 requirements_demo.txt | 11 +++++++++++
 4 files changed, 37 insertions(+), 11 deletions(-)
 create mode 100644 requirements.txt
 create mode 100644 requirements_demo.txt

diff --git a/README.md b/README.md
index df9fb4b..32e11fa 100644
--- a/README.md
+++ b/README.md
@@ -35,12 +35,17 @@ First, clone this repository to your local machine, and install the dependencies
 ```bash
 git clone git@github.com:facebookresearch/vggt.git 
 cd vggt
+pip install -r requirements.txt
 ```
+
+<details>
+<summary>Alternatively, you can install VGGT as a package via pip, pixi or uv (Click to expand)</summary>
+
 ### Install with pip
+
 ```bash
 pip install -e .
 ```
-
 ### Install and run with pixi
 Download [pixi here](https://pixi.sh)
 
@@ -53,6 +58,8 @@ Download [uv here](https://docs.astral.sh/uv/getting-started/installation/)
 ```bash
 uv run --extra demo demo_gradio.py
 ```
+</details>
+
 
 Now, try the model with just a few lines of code:
 
@@ -243,4 +250,4 @@ Thanks to these great repositories: [PoseDiffusion](https://github.com/facebookr
 
 
 ## License
-See the [LICENSE](./LICENSE.txt) file for details about the license under which this code is made available.
+See the [LICENSE](./LICENSE.txt) file for details about the license under which this code is made available.
\ No newline at end of file
diff --git a/pyproject.toml b/pyproject.toml
index dda64e8..c019a96 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,17 +1,15 @@
 [project]
-authors = [{ name = "Jianyuan Wang", email = "jianyuan@robots.ox.ac.uk" }]
+authors = [{name = "Jianyuan Wang", email = "jianyuan@robots.ox.ac.uk"}]
 dependencies = [
-    "torch>=2.3.1",
-    "torchvision>=0.18.1",
-    "numpy<2",
+    "numpy==1.26.1",
     "Pillow",
     "huggingface_hub",
     "einops",
     "safetensors",
 ]
 name = "vggt"
-requires-python = ">= 3.11"
-version = "0.1.0"
+requires-python = ">= 3.8"
+version = "0.0.1"
 
 [project.optional-dependencies]
 demo = [
@@ -28,13 +26,16 @@ demo = [
     "matplotlib",
 ]
 
+# Using setuptools as the build backend
 [build-system]
-build-backend = "hatchling.build"
-requires = ["hatchling"]
+requires = ["setuptools>=61.0", "wheel"]
+build-backend = "setuptools.build_meta"
 
-[tool.hatch.build.targets.wheel]
+# setuptools configuration
+[tool.setuptools]
 packages = ["vggt"]
 
+# Pixi configuration
 [tool.pixi.workspace]
 channels = ["conda-forge"]
 platforms = ["linux-64"]
diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 0000000..1cb61a2
--- /dev/null
+++ b/requirements.txt
@@ -0,0 +1,7 @@
+torch==2.3.1
+torchvision==0.18.1
+numpy==1.26.1
+Pillow
+huggingface_hub
+einops
+safetensors
diff --git a/requirements_demo.txt b/requirements_demo.txt
new file mode 100644
index 0000000..efd6f5e
--- /dev/null
+++ b/requirements_demo.txt
@@ -0,0 +1,11 @@
+gradio==5.17.1
+viser==0.2.23
+tqdm
+hydra-core
+omegaconf
+opencv-python
+scipy
+onnxruntime
+requests
+trimesh
+matplotlib