Upload 2 files

app.py

@@ -4,53 +4,19 @@ import tempfile
 import threading
 from pathlib import Path
 
-# === Runtime environment tweaks (must be set before importing heavy libs) ===
+# === Runtime environment tweaks ===
 os.environ.setdefault("CUDA_VISIBLE_DEVICES", "")  # force CPU
-os.environ.setdefault("WORLDGEN_DISABLE_EVAL", "1")  # avoid KNN/eval paths if supported
-os.environ.setdefault("GRADIO_TEMP_DIR", "/tmp/gradio")  # centralize gradio cache
-
-import nunchaku.utils
-
-def safe_get_precision(*args, **kwargs):
-    # Always return fp32 on CPU to avoid any torch.cuda references
-    return "fp32"
-
-# Patch module attribute
-nunchaku.utils.get_precision = safe_get_precision
-
-# ALSO patch the module that imported the symbol name directly
-try:
-    import nunchaku.models.transformers.transformer_flux as tf
-    tf.get_precision = safe_get_precision
-except Exception:
-    # If the module isn't imported yet, it will still use our utils patch when it imports later.
-    pass
+os.environ.setdefault("WORLDGEN_DISABLE_EVAL", "1")
+os.environ.setdefault("GRADIO_TEMP_DIR", "/tmp/gradio")
 
 import gradio as gr
-import torch
 from worldgen import WorldGen
 
-from huggingface_hub import hf_hub_download
-
 _wg = None
-
 def get_worldgen():
     global _wg
     if _wg is None:
-        # 確保使用公開模型且不帶 token
-        model_path = hf_hub_download(
-            repo_id="nunchaku-tech/nunchaku-flux.1-dev",
-            filename="unquantized_layers.safetensors",
-            token=None  # 強制匿名模式
-        )
-
-        # 初始化 WorldGen，並指定模型路徑（如果需要）
-        _wg = WorldGen(
-            mode="t2s",
-            device="cpu",
-            low_vram=True,
-            lora_path=None  # 如果你有 LoRA 模型可以指定路徑
-        )
+        _wg = WorldGen(mode="t2s", device="cpu", low_vram=True, lora_path=None)
     return _wg
 
 def _zip_directory(dir_path: Path) -> str:
@@ -62,21 +28,15 @@ def _zip_directory(dir_path: Path) -> str:
 
 def _save_scene(generated, out_dir: Path) -> Path:
     out_dir.mkdir(parents=True, exist_ok=True)
-
-    # 1) try common object methods
     for meth in ("save", "export", "write", "to_disk"):
         if hasattr(generated, meth):
             getattr(generated, meth)(str(out_dir))
             return out_dir
-
-    # 2) single-file export fallbacks
     for meth in ("save_as_glb", "to_glb", "save_as_ply", "to_ply"):
         if hasattr(generated, meth):
             fp = out_dir / ("scene.glb" if "glb" in meth else "scene.ply")
             getattr(generated, meth)(str(fp))
             return out_dir
-
-    # 3) last resort: leave a note
     (out_dir / "README.txt").write_text(
         "No known save/export method detected. Please adjust _save_scene according to your WorldGen version."
     )
@@ -85,50 +45,31 @@ def _save_scene(generated, out_dir: Path) -> Path:
 def generate_from_text(prompt: str):
     if not prompt or not prompt.strip():
         return None, "Please enter a scene description (English works best)."
-
     wg = get_worldgen()
-
     try:
         result = wg.generate_world(prompt.strip())
     except Exception as e:
         return None, f"Generation failed: {e}"
-
     tmpdir = Path(tempfile.mkdtemp())
     out_dir = tmpdir / "worldgen_output"
     out_dir = _save_scene(result, out_dir)
     zip_path = _zip_directory(out_dir)
-
-    # delayed cleanup to avoid filling the 50GB ephemeral disk
     def _cleanup(p: Path):
         try:
             shutil.rmtree(p, ignore_errors=True)
         except Exception:
             pass
-
-    threading.Timer(300, _cleanup, args=(tmpdir,)).start()  # 5 minutes later
-
+    threading.Timer(300, _cleanup, args=(tmpdir,)).start()
     return zip_path, "Done. Download the ZIP (contains 3D assets)."
 
 with gr.Blocks(title="WorldGen • Text -> 3D Scene") as demo:
-    gr.Markdown(
-        """# WorldGen - Text -> 3D
-- CPU-only fallback is enabled. For real runs, switch the Space to a GPU (T4/A10G).
-- Outputs are zipped for download; temp files are cleaned shortly after each run.
-
-"""
-    )
+    gr.Markdown("# WorldGen - Text -> 3D\n- CPU-only fallback is enabled.\n- Outputs are zipped for download.")
     with gr.Row():
         with gr.Column():
-            prompt = gr.Textbox(
-                label="Scene prompt (English)",
-                placeholder="a neon-lit cyberpunk street with rain reflections, highly detailed",
-                lines=3,
-            )
+            prompt = gr.Textbox(label="Scene prompt (English)", placeholder="a neon-lit cyberpunk street", lines=3)
             btn = gr.Button("Generate 3D Scene")
         with gr.Column():
             out_zip = gr.File(label="Download output (ZIP)")
             status = gr.Markdown()
-
     btn.click(generate_from_text, inputs=[prompt], outputs=[out_zip, status])
-
 demo.launch()

requirements.txt

@@ -1,4 +1,6 @@
 torch>=2.7
-# torchvision>=0.22  # uncomment if needed by your WorldGen setup
 websockets>=13.1
+gradio
+huggingface_hub
+nunchaku
 git+https://github.com/ZiYang-xie/WorldGen.git