Hugging Face's logo

Spaces:
hzxie
/
city-dreamer
Running on A10G

App Files Community
hzxie commited on
Commit
f8ec29a
•
1 Parent(s): 79df973

fix: runtime environment is not set up.

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,7 +4,7 @@
4
  # @Author: Haozhe Xie
5
  # @Date: 2024-03-02 16:30:00
6
  # @Last Modified by: Haozhe Xie
7
- # @Last Modified at: 2024-03-03 12:02:23
8
  # @Email: root@haozhexie.com
9
 
10
  import gradio as gr
@@ -105,7 +105,7 @@ if __name__ == "__main__":
105
  format="[%(levelname)s] %(asctime)s %(message)s", level=logging.INFO
106
  )
107
  logging.info("Compiling CUDA extensions...")
108
- # setup_runtime_env()
109
 
110
  logging.info("Downloading pretrained models...")
111
  fgm = get_models("CityDreamer-Fgnd.pth")
 
4
  # @Author: Haozhe Xie
5
  # @Date: 2024-03-02 16:30:00
6
  # @Last Modified by: Haozhe Xie
7
+ # @Last Modified at: 2024-03-03 12:21:20
8
  # @Email: root@haozhexie.com
9
 
10
  import gradio as gr
 
105
  format="[%(levelname)s] %(asctime)s %(message)s", level=logging.INFO
106
  )
107
  logging.info("Compiling CUDA extensions...")
108
+ setup_runtime_env()
109
 
110
  logging.info("Downloading pretrained models...")
111
  fgm = get_models("CityDreamer-Fgnd.pth")