chahui commited on
Commit
5fc5dd3
·
verified ·
1 Parent(s): 789c23e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -0
app.py CHANGED
@@ -1,3 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
  import shutil, tempfile
2
  from pathlib import Path
3
  import gradio as gr, torch
 
1
+ import os
2
+ os.environ["CUDA_VISIBLE_DEVICES"] = "" # 強制 CPU
3
+ os.environ["WORLDGEN_DISABLE_EVAL"] = "1" # 若程式支援,避免 KNN
4
+
5
+ import gradio as gr
6
+ from worldgen import WorldGen
7
+
8
+ wg = WorldGen(mode="t2s", device="cpu", low_vram=True)
9
+
10
+ gr.Markdown("""# WorldGen — 文字 -> 3D 場景- 僅啟用 Text -> 3D。- Free CPU 會很慢,建議升級 GPU(T4/A10G)。""")
11
+ # 其餘邏輯同前;記得回傳後用延時 Thread 清理臨時資料夾,避免 50GB 被用滿
12
+
13
  import shutil, tempfile
14
  from pathlib import Path
15
  import gradio as gr, torch