File size: 805 Bytes
7337389
 
 
 
b8740b1
7337389
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
"""Tải sẵn trọng số mô hình lúc build image để Space khởi động nhanh hơn."""
import os
from huggingface_hub import snapshot_download

tok = os.environ.get("HF_TOKEN") or None   # chuỗi rỗng làm hỏng header Authorization
for name in (os.environ.get("RETRIEVER_MODEL", "bkai-foundation-models/vietnamese-bi-encoder"),
             os.environ.get("READER_MODEL", "deepset/xlm-roberta-base-squad2")):
    try:
        snapshot_download(name, token=tok,
                          ignore_patterns=["*.onnx", "*.h5", "*.msgpack", "*.ot", "onnx/*",
                                           "openvino/*", "*.tflite", "rust_model*"])
        print("preloaded", name)
    except Exception as e:  # không chặn build; lúc chạy sẽ tải lại
        print("skip preload", name, e)