Try to migrate to ZERO.

README.md

@@ -4,7 +4,7 @@ emoji: 🏙️
 colorFrom: blue
 colorTo: green
 sdk: gradio
-sdk_version: 4.19.2
+sdk_version: 4.44.0
 app_file: app.py
 pinned: false
 license: other

app.py

@@ -4,13 +4,14 @@
 # @Author: Haozhe Xie
 # @Date:   2024-03-02 16:30:00
 # @Last Modified by: Haozhe Xie
-# @Last Modified at: 2024-03-03 16:08:25
+# @Last Modified at: 2024-09-21 10:36:47
 # @Email:  [email protected]
 
 import gradio as gr
 import logging
 import numpy as np
 import os
+import spaces
 import ssl
 import subprocess
 import sys
@@ -61,7 +62,10 @@ def get_city_layout():
     return hf, seg
 
 
-def get_generated_city(radius, altitude, azimuth, map_center):
+@spaces.GPU
+def get_generated_city(
+    radius, altitude, azimuth, map_center, progress=gr.Progress(track_tqdm=True)
+):
     # The import must be done after CUDA extension compilation
     import citydreamer.inference
 

citydreamer/extensions/grid_encoder/setup.py

@@ -22,10 +22,10 @@ setup(
                 "bindings.cpp",
             ],
             extra_compile_args={
-                "cxx": ["-O3", "-std=c++14"],
+                "cxx": ["-O3", "-std=c++17"],
                 "nvcc": [
                     "-O3",
-                    "-std=c++14",
+                    "-std=c++17",
                     "-U__CUDA_NO_HALF_OPERATORS__",
                     "-U__CUDA_NO_HALF_CONVERSIONS__",
                     "-U__CUDA_NO_HALF2_OPERATORS__",

citydreamer/model.py

@@ -610,9 +610,11 @@ class RenderMLP(torch.nn.Module):
                 in_dim = f_dim
 
         self.fc_m_a = torch.nn.Linear(
-            cfg.NETWORK.GANCRAFT.N_CLASSES + 1
-            if cfg.NETWORK.GANCRAFT.BUILDING_MODE
-            else cfg.NETWORK.GANCRAFT.N_CLASSES,
+            (
+                cfg.NETWORK.GANCRAFT.N_CLASSES + 1
+                if cfg.NETWORK.GANCRAFT.BUILDING_MODE
+                else cfg.NETWORK.GANCRAFT.N_CLASSES
+            ),
             cfg.NETWORK.GANCRAFT.RENDER_HIDDEN_DIM,
             bias=False,
         )

requirements.txt

@@ -1,5 +1,5 @@
---extra-index-url https://download.pytorch.org/whl/cu113
-torch==1.12.0
+--extra-index-url https://download.pytorch.org/whl/cu118
+torch==2.4.1
 torchvision
 
 easydict