Spaces:
Runtime error
Runtime error
XiaoyiYangRIT
commited on
Commit
·
a6cd9f8
1
Parent(s):
62b60d1
Update some files
Browse files
app.py
CHANGED
|
@@ -3,6 +3,7 @@ import torch
|
|
| 3 |
import math
|
| 4 |
import os
|
| 5 |
from transformers import AutoTokenizer, AutoModel, AutoProcessor
|
|
|
|
| 6 |
from decord import VideoReader, cpu
|
| 7 |
from PIL import Image
|
| 8 |
from torchvision.transforms import Compose, Resize, ToTensor, Normalize
|
|
@@ -23,9 +24,10 @@ MODEL_NAME = "OpenGVLab/InternVL3-14B"
|
|
| 23 |
|
| 24 |
# 如果第一次运行:下载模型并缓存到 /data
|
| 25 |
if not os.path.exists(PERSISTENT_DIR):
|
| 26 |
-
print("
|
| 27 |
-
from transformers import snapshot_download
|
| 28 |
snapshot_download(repo_id=MODEL_NAME, local_dir=PERSISTENT_DIR, trust_remote_code=True)
|
|
|
|
|
|
|
| 29 |
|
| 30 |
# 模型加载(从本地)
|
| 31 |
tokenizer = AutoTokenizer.from_pretrained(PERSISTENT_DIR, trust_remote_code=True)
|
|
|
|
| 3 |
import math
|
| 4 |
import os
|
| 5 |
from transformers import AutoTokenizer, AutoModel, AutoProcessor
|
| 6 |
+
from huggingface_hub import snapshot_download
|
| 7 |
from decord import VideoReader, cpu
|
| 8 |
from PIL import Image
|
| 9 |
from torchvision.transforms import Compose, Resize, ToTensor, Normalize
|
|
|
|
| 24 |
|
| 25 |
# 如果第一次运行:下载模型并缓存到 /data
|
| 26 |
if not os.path.exists(PERSISTENT_DIR):
|
| 27 |
+
print("⏬ First run: downloading model to persistent storage...")
|
|
|
|
| 28 |
snapshot_download(repo_id=MODEL_NAME, local_dir=PERSISTENT_DIR, trust_remote_code=True)
|
| 29 |
+
else:
|
| 30 |
+
print("✅ Loaded model from persistent cache.")
|
| 31 |
|
| 32 |
# 模型加载(从本地)
|
| 33 |
tokenizer = AutoTokenizer.from_pretrained(PERSISTENT_DIR, trust_remote_code=True)
|