update

app.py

@@ -45,7 +45,10 @@ import pytorch_lightning as pl
 
 MAX_SEED = np.iinfo(np.int32).max
 TMP_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'tmp')
+WEIGHTS_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'weights')
 os.makedirs(TMP_DIR, exist_ok=True)
+os.makedirs(WEIGHTS_DIR, exist_ok=True)
+
 
 
 
@@ -56,6 +59,7 @@ def cache_weights(weights_dir: str) -> dict:
     os.makedirs(weights_dir, exist_ok=True)
     model_ids = [
         "Stable-X/trellis-normal-v0-1",
+        "houyuanchen/lino"
     ]
     cached_paths = {}
     for model_id in model_ids:
@@ -370,7 +374,7 @@ with gr.Blocks(css="footer {visibility: hidden}") as demo:
 
 if __name__ == "__main__":
     # Download and cache the weights
-    #cache_weights(WEIGHTS_DIR)
+    cache_weights(WEIGHTS_DIR)
 
     hi3dgen_pipeline = Hi3DGenPipeline.from_pretrained("weights/trellis-normal-v0-1")
     hi3dgen_pipeline.cuda()

requirements.txt

@@ -1,8 +1,12 @@
+torch==2.4.0+cu124
+torchmetrics==1.7.3
+torchvision==0.19.0+cu124
+pytorch_lightning==2.5.1.post0
+
 diffusers==0.33.1
 einops==0.8.1
-flash_attn==2.8.0.post2
 gradio==5.34.0
-huggingface_hub==0.33.0
+
 hydra-core==1.3.2
 lightning==2.5.1.post0
 lightning_utilities==0.14.3
@@ -11,18 +15,17 @@ numpy==2.3.0
 omegaconf==2.3.0
 opencv_python==4.11.0.86
 Pillow==11.2.1
-pytorch_lightning==2.5.1.post0
+
 rich==14.0.0
 safetensors==0.5.3
 scikit_learn==1.7.0
 scipy==1.15.3
 skimage==0.0
 spconv_cu124==2.3.8
-torch==2.4.0+cu124
-torchmetrics==1.7.3
-torchvision==0.19.0+cu124
 tqdm==4.67.1
 transformers==4.46.3
 trimesh==4.6.12
 wandb==0.20.1
 xformers==0.0.27.post2
+flash_attn==2.8.0.post2
+huggingface_hub==0.33.0