mount bucket storage

app.py

@@ -8,17 +8,28 @@ import spaces
 
 import os
 # Hugging Face 相关缓存
-os.environ["HF_HOME"] = "/data/.huggingface"
-os.environ["HF_HUB_CACHE"] = "/data/.cache/huggingface/hub"
-os.environ["HF_DATASETS_CACHE"] = "/data/.cache/huggingface/datasets"
-os.environ["TRANSFORMERS_CACHE"] = "/data/.cache/huggingface/hub"
+os.environ["HF_HOME"] = "/cache_data/.huggingface"
+os.environ["HF_HUB_CACHE"] = "/cache_data/.cache/huggingface/hub"
+os.environ["HF_DATASETS_CACHE"] = "/cache_data/.cache/huggingface/datasets"
+os.environ["TRANSFORMERS_CACHE"] = "/cache_data/.cache/huggingface/hub"
 
 # PyTorch / Torch Hub / TorchVision / 部分 timm 权重
-os.environ["TORCH_HOME"] = "/data/.cache/torch"
-os.environ["XDG_CACHE_HOME"] = "/data/.cache"
+os.environ["TORCH_HOME"] = "/cache_data/.cache/torch"
+os.environ["XDG_CACHE_HOME"] = "/cache_data/.cache"
 
 # 关键：PyTorch JIT CUDA/C++ 扩展编译缓存目录
-os.environ["TORCH_EXTENSIONS_DIR"] = "/data/.cache/torch_extensions"
+os.environ["TORCH_EXTENSIONS_DIR"] = "/cache_data/.cache/torch_extensions"
+
+os.environ["TMPDIR"] = "/cache_data/bucket/tmp"
+
+os.makedirs(os.environ["HF_HOME"], exists_ok=True)
+os.makedirs(os.environ["HF_HUB_CACHE"], exists_ok=True)
+os.makedirs(os.environ["HF_DATASETS_CACHE"], exists_ok=True)
+os.makedirs(os.environ["TRANSFORMERS_CACHE"], exists_ok=True)
+os.makedirs(os.environ["TORCH_HOME"], exists_ok=True)
+os.makedirs(os.environ["XDG_CACHE_HOME"], exists_ok=True)
+os.makedirs(os.environ["TORCH_EXTENSIONS_DIR"], exists_ok=True)
+os.makedirs(os.environ["TMPDIR"], exists_ok=True)
 
 import uuid
 from typing import Any, List, Optional, Union
@@ -59,7 +70,8 @@ generated_object_map = {}
 
 ############## 3D-Fixer model
 model_dir = 'HorizonRobotics/3D-Fixer'
-local_dir = "/data/checkpoints/3D-Fixer"
+local_dir = "/cache_data/checkpoints/3D-Fixer"
+os.makedirs(local_dir, exist_ok=True)
 snapshot_download(repo_id=model_dir, local_dir=local_dir)
 ############## 3D-Fixer model