Fix some bugs.

.gitattributes

@@ -20,7 +20,6 @@
 *.pickle filter=lfs diff=lfs merge=lfs -text
 *.pkl filter=lfs diff=lfs merge=lfs -text
 *.pt filter=lfs diff=lfs merge=lfs -text
-*.pth filter=lfs diff=lfs merge=lfs -text
 *.rar filter=lfs diff=lfs merge=lfs -text
 *.safetensors filter=lfs diff=lfs merge=lfs -text
 saved_model/**/* filter=lfs diff=lfs merge=lfs -text
@@ -34,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
 *.png filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.whl filter=lfs diff=lfs merge=lfs -text

README.md

@@ -7,7 +7,6 @@ sdk: gradio
 sdk_version: 4.44.0
 app_file: app.py
 pinned: false
-license: other
 ---
 
 Official demo for **[CityDreamer: Compositional Generative Model of Unbounded 3D Cities](https://github.com/hzxie/city-dreamer) (CVPR 2024).**

app.py

@@ -27,15 +27,23 @@ sys.path.append(os.path.join(os.path.dirname(__file__), "citydreamer"))
 
 
 def setup_runtime_env():
-    logging.info("CUDA version is %s" % subprocess.check_output(["nvcc", "--version"]))
+    try:
+        logging.info("CUDA version is %s" % subprocess.check_output(["nvcc", "--version"]))
+    except Exception as ex:
+        logging.exception(ex)
+
     logging.info("GCC version is %s" % subprocess.check_output(["g++", "--version"]))
-    # Compile CUDA extensions
-    ext_dir = os.path.join(os.path.dirname(__file__), "citydreamer", "extensions")
+    # Instsall Pre-compiled CUDA extensions
+    ext_dir = os.path.join(os.path.dirname(__file__), "wheels")
     for e in os.listdir(ext_dir):
-        if not os.path.isdir(os.path.join(ext_dir, e)):
-            continue
+        if not os.path.join(ext_dir, e).endswith(".whl"):
+            subprocess.call(["pip", "install", os.path.join(ext_dir, e)])
 
-        subprocess.call(["pip", "install", "."], cwd=os.path.join(ext_dir, e))
+    # Compile CUDA extensions
+    # ext_dir = os.path.join(os.path.dirname(__file__), "citydreamer", "extensions")
+    # for e in os.listdir(ext_dir):
+    #     if os.path.isdir(os.path.join(ext_dir, e)):
+    #         subprocess.call(["pip", "install", "."], cwd=os.path.join(ext_dir, e))
 
 
 def get_models(file_name):

requirements.txt

@@ -1,10 +1,9 @@
 --extra-index-url https://download.pytorch.org/whl/cu118
-torch==2.4.1
+torch==2.2.2
 torchvision
 
 easydict
 gradio
-numpy
+numpy<2.0.0
 opencv-python
 pillow
-

wheels/extrude_tensor-1.0.0-cp310-cp310-linux_x86_64.whl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:58b87e506ae559cb11e6cb1375bdc6c119f02f3d9560b015b46d3561048b3fd2
+size 237463

wheels/grid_encoder-1.0.0-cp310-cp310-linux_x86_64.whl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8b271befcb1ed3603cf0d62d0599ad7288a70886c98a66b350fdf4a64529ed3f
+size 2375579

wheels/voxrender-1.0.0-cp310-cp310-linux_x86_64.whl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e65240d2379e48f9e4a541b146ad9b3f66bc0b6549c0732bfc2aa63bb7abc7eb
+size 593162