Spaces:

tencent
/

Hunyuan3D-2

Running on Zero

App Files Files Community

Huiwenshi commited on 2 days ago

Commit

60a23f4

verified ·

1 Parent(s): e21b138

Upload folder using huggingface_hub

Browse files

Files changed (8) hide show

.gitattributes +1 -0
gradio_cache/0/input.png +0 -0
gradio_cache/0/rembg.png +0 -0
gradio_cache/0/textured_mesh.glb +3 -0
gradio_cache/0/textured_mesh.html +40 -0
gradio_cache/0/white_mesh.glb +0 -0
gradio_cache/0/white_mesh.html +57 -0
hg_app.py +41 -25

.gitattributes CHANGED Viewed

@@ -38,3 +38,4 @@ assets/images/e2e-1.gif filter=lfs diff=lfs merge=lfs -text
 assets/images/e2e-2.gif filter=lfs diff=lfs merge=lfs -text
 assets/images/system.jpg filter=lfs diff=lfs merge=lfs -text
 assets/images/teaser.jpg filter=lfs diff=lfs merge=lfs -text

 assets/images/e2e-2.gif filter=lfs diff=lfs merge=lfs -text
 assets/images/system.jpg filter=lfs diff=lfs merge=lfs -text
 assets/images/teaser.jpg filter=lfs diff=lfs merge=lfs -text
+gradio_cache/0/textured_mesh.glb filter=lfs diff=lfs merge=lfs -text

gradio_cache/0/input.png ADDED Viewed

gradio_cache/0/rembg.png ADDED Viewed

gradio_cache/0/textured_mesh.glb ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:e264d411d95bf5b967418ca0bf185a22897d2d3c172d169830269b9d4813bcb4
+size 2531000

gradio_cache/0/textured_mesh.html ADDED Viewed

	@@ -0,0 +1,40 @@

+<!DOCTYPE html>
+<html>
+<head>
+    <!-- Import the component -->
+    <script src="https://ajax.googleapis.com/ajax/libs/model-viewer/3.1.1/model-viewer.min.js" type="module"></script>
+    <style>
+        body {
+            margin: 0;
+            font-family: Arial, sans-serif;
+        }
+        .centered-container {
+            display: flex;
+            justify-content: center;
+            align-items: center;
+            border-radius: 8px;
+            border-color: #e5e7eb;
+            border-style: solid;
+            border-width: 1px;
+        }
+    </style>
+</head>
+<body>
+<div class="centered-container">
+            <div class="column is-mobile is-centered">
+                <model-viewer style="height: 586px; width: 700px;" rotation-per-second="10deg" id="modelViewer"
+                    src="./textured_mesh.glb/" disable-tap
+                    environment-image="neutral" auto-rotate camera-target="0m 0m 0m" orientation="0deg 0deg 170deg" shadow-intensity=".9"
+                    ar auto-rotate camera-controls>
+                </model-viewer>
+            </div>
+</div>
+</body>
+</html>

gradio_cache/0/white_mesh.glb ADDED Viewed

Binary file (723 kB). View file

gradio_cache/0/white_mesh.html ADDED Viewed

	@@ -0,0 +1,57 @@

+<!DOCTYPE html>
+<html>
+<head>
+    <!-- Import the component -->
+    <script src="https://ajax.googleapis.com/ajax/libs/model-viewer/3.1.1/model-viewer.min.js" type="module"></script>
+    <script>
+        document.addEventListener('DOMContentLoaded', () => {
+            const modelViewers = document.querySelectorAll('model-viewer');
+            modelViewers.forEach(modelViewer => {
+                modelViewer.addEventListener('load', (event) => {
+                    const [material] = modelViewer.model.materials;
+                    let color = [43, 44, 46, 255];
+                    color = color.map(x => x / 255);
+                    material.pbrMetallicRoughness.setMetallicFactor(0.1); // 完全金属
+                    material.pbrMetallicRoughness.setRoughnessFactor(0.7); // 低粗糙度
+                    material.pbrMetallicRoughness.setBaseColorFactor(color);  // CornflowerBlue in RGB
+                });
+            });
+        });
+    </script>
+    <style>
+        body {
+            margin: 0;
+            font-family: Arial, sans-serif;
+        }
+        .centered-container {
+            display: flex;
+            justify-content: center;
+            align-items: center;
+            border-radius: 8px;
+            border-color: #e5e7eb;
+            border-style: solid;
+            border-width: 1px;
+        }
+    </style>
+</head>
+<body>
+<div class="centered-container">
+            <div class="column is-mobile is-centered">
+                <model-viewer style="height: 586px; width: 700px;" rotation-per-second="10deg" id="modelViewer"
+                    src="./white_mesh.glb/" disable-tap
+                    environment-image="neutral" auto-rotate camera-target="0m 0m 0m" orientation="0deg 0deg 170deg" shadow-intensity=".9"
+                    ar auto-rotate camera-controls>
+                </model-viewer>
+            </div>
+</div>
+</body>
+</html>

hg_app.py CHANGED Viewed

@@ -1,26 +1,42 @@
-import os
-import spaces
-import subprocess
-def install_cuda_toolkit():
-    # CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run"
-    CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/12.2.0/local_installers/cuda_12.2.0_535.54.03_linux.run"
-    CUDA_TOOLKIT_FILE = "/tmp/%s" % os.path.basename(CUDA_TOOLKIT_URL)
-    subprocess.call(["wget", "-q", CUDA_TOOLKIT_URL, "-O", CUDA_TOOLKIT_FILE])
-    subprocess.call(["chmod", "+x", CUDA_TOOLKIT_FILE])
-    subprocess.call([CUDA_TOOLKIT_FILE, "--silent", "--toolkit"])
-    os.environ["CUDA_HOME"] = "/usr/local/cuda"
-    os.environ["PATH"] = "%s/bin:%s" % (os.environ["CUDA_HOME"], os.environ["PATH"])
-    os.environ["LD_LIBRARY_PATH"] = "%s/lib:%s" % (
-        os.environ["CUDA_HOME"],
-        "" if "LD_LIBRARY_PATH" not in os.environ else os.environ["LD_LIBRARY_PATH"],
-    )
-    # Fix: arch_list[-1] += '+PTX'; IndexError: list index out of range
-    os.environ["TORCH_CUDA_ARCH_LIST"] = "8.0;8.6"
-install_cuda_toolkit()
-os.system("cd /home/user/app/hy3dgen/texgen/differentiable_renderer/ && bash compile_mesh_painter.sh")
-os.system("cd /home/user/app/hy3dgen/texgen/custom_rasterizer && pip install .")
 import os
 import shutil
@@ -412,6 +428,6 @@ if __name__ == '__main__':
     app.mount("/static", StaticFiles(directory=static_dir), name="static")
     demo = build_app()
-    demo.queue(max_size=3)
     app = gr.mount_gradio_app(app, demo, path="/")
-    uvicorn.run(app, host="0.0.0.0", port=7860)

+# pip install gradio==4.44.1
+if True:
+    import os
+    import spaces
+    import subprocess
+    def install_cuda_toolkit():
+        # CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run"
+        CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/12.2.0/local_installers/cuda_12.2.0_535.54.03_linux.run"
+        CUDA_TOOLKIT_FILE = "/tmp/%s" % os.path.basename(CUDA_TOOLKIT_URL)
+        subprocess.call(["wget", "-q", CUDA_TOOLKIT_URL, "-O", CUDA_TOOLKIT_FILE])
+        subprocess.call(["chmod", "+x", CUDA_TOOLKIT_FILE])
+        subprocess.call([CUDA_TOOLKIT_FILE, "--silent", "--toolkit"])
+        os.environ["CUDA_HOME"] = "/usr/local/cuda"
+        os.environ["PATH"] = "%s/bin:%s" % (os.environ["CUDA_HOME"], os.environ["PATH"])
+        os.environ["LD_LIBRARY_PATH"] = "%s/lib:%s" % (
+            os.environ["CUDA_HOME"],
+            "" if "LD_LIBRARY_PATH" not in os.environ else os.environ["LD_LIBRARY_PATH"],
+        )
+        # Fix: arch_list[-1] += '+PTX'; IndexError: list index out of range
+        os.environ["TORCH_CUDA_ARCH_LIST"] = "8.0;8.6"
+    # install_cuda_toolkit()
+    os.system("cd /home/user/app/hy3dgen/texgen/differentiable_renderer/ && bash compile_mesh_painter.sh")
+    os.system("cd /home/user/app/hy3dgen/texgen/custom_rasterizer && pip install .")
+    # os.system("cd /home/user/app/hy3dgen/texgen/custom_rasterizer && CUDA_HOME=/usr/local/cuda FORCE_CUDA=1 TORCH_CUDA_ARCH_LIST='8.0;8.6;8.9;9.0' python setup.py install")
+    IP = "0.0.0.0"
+    PORT = 7860
+else:
+    IP = "0.0.0.0"
+    PORT = 8080
+    class spaces:
+        class GPU:
+            def __init__(self, duration=60):
+                self.duration = duration
+            def __call__(self, func):
+                return func
 import os
 import shutil
     app.mount("/static", StaticFiles(directory=static_dir), name="static")
     demo = build_app()
+    demo.queue(max_size=4)
     app = gr.mount_gradio_app(app, demo, path="/")
+    uvicorn.run(app, host=IP, port=PORT)