Upload app_hg.py with huggingface_hub

app_hg.py

@@ -39,17 +39,6 @@ import sys
 import subprocess
 from glob import glob
 
-# @spaces.GPU
-# def check_env():
-#     print(glob("/usr/local/cuda/*"))
-#     print(torch.cuda.is_available())
-#     print(os.environ.get('CUDA_HOME', None))
-#     os.environ['CUDA_HOME'] = '/usr/local/cuda'
-#     # Optionally, update PATH and LD_LIBRARY_PATH if needed
-#     os.environ['PATH'] = os.environ['CUDA_HOME'] + '/bin:' + os.environ['PATH']
-#     os.environ['LD_LIBRARY_PATH'] = os.environ['CUDA_HOME'] + '/lib64:' + os.environ.get('LD_LIBRARY_PATH', '')
-# check_env()
-
 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"
@@ -66,18 +55,9 @@ def install_cuda_toolkit():
     )
     # Fix: arch_list[-1] += '+PTX'; IndexError: list index out of range
     os.environ["TORCH_CUDA_ARCH_LIST"] = "8.0;8.6"
-    try:
-        subprocess.check_call([sys.executable, "-m", "pip", "install", "-U", "git+https://github.com/facebookresearch/pytorch3d@stable"])
-        import pytorch3d
-    except Exception as err:
-        print(err)
-        print("new inner cuda install pytorch3d fail")
+
 install_cuda_toolkit()
 
-# def install_requirements():
-#     # Install the packages listed in requirements.txt
-#     subprocess.check_call([sys.executable, "-m", "pip", "install", "-r", "requirements.txt"])
-#install_requirements()
 
 from infer import seed_everything, save_gif
 from infer import Text2Image, Removebg, Image2Views, Views2Mesh, GifRenderer
@@ -92,7 +72,7 @@ parser.add_argument("--use_lite", default=False, action="store_true")
 parser.add_argument("--mv23d_cfg_path", default="./svrm/configs/svrm.yaml", type=str)
 parser.add_argument("--mv23d_ckt_path", default="weights/svrm/svrm.safetensors", type=str)
 parser.add_argument("--text2image_path", default="weights/hunyuanDiT", type=str)
-parser.add_argument("--save_memory", default=False, action="store_true")
+parser.add_argument("--save_memory", default=True)
 parser.add_argument("--device", default="cuda:0", type=str)
 args = parser.parse_args()
 
@@ -365,7 +345,7 @@ with gr.Blocks() as demo:
                         textgen_SEED = gr.Number(value=0, label="Gen seed", precision=0, interactive=True)
                         textgen_STEP = gr.Number(value=50, label="Gen steps", precision=0, 
                                                  minimum=40, maximum=100, interactive=True)
-                        textgen_max_faces = gr.Number(value=90000, label="Face number", precision=0, 
+                        textgen_max_faces = gr.Number(value=60000, label="Face number", precision=0, 
                                                       minimum=5000, maximum=1000000, interactive=True)
                     with gr.Row():
                         textgen_submit = gr.Button("Generate", variant="primary")
@@ -398,7 +378,7 @@ with gr.Blocks() as demo:
                     imggen_SEED = gr.Number(value=0, label="Gen seed", precision=0, interactive=True)
                     imggen_STEP = gr.Number(value=50, label="Gen steps", precision=0, 
                                             minimum=40, maximum=100, interactive=True)
-                    imggen_max_faces = gr.Number(value=90000, label="Face number", precision=0, 
+                    imggen_max_faces = gr.Number(value=60000, label="Face number", precision=0, 
                                                      minimum=5000, maximum=1000000, interactive=True)
                 with gr.Row():
                     imggen_submit = gr.Button("Generate", variant="primary")