File size: 1,018 Bytes
e8b4369 8fa33c0 fcae93c 22dcdf9 0b92c2a 22dcdf9 8903c9f e0bcf49 8903c9f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
# app.py
import os
from fastapi import FastAPI
from huggingface_hub import login
app = FastAPI()
@app.get("/")
def root():
login(os.environ.get("HUGGINGFACE_HUB_TOKEN", ""))
return {"status": "ok"}
# app.py
import os
from fastapi import FastAPI
from huggingface_hub import hf_hub_download
# 必ず import の前に環境変数を設定！
os.environ["HF_HOME"] = "/tmp/hf"
os.environ["HF_HUB_CACHE"] = "/tmp/hf/hub"
os.environ["HUGGINGFACE_HUB_CACHE"] = "/tmp/hf/hub"
os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf/transformers"
os.environ["XDG_CACHE_HOME"] = "/tmp/hf/xdg"
app = FastAPI()
@app.get("/")
def root():
model_path = hf_hub_download(
repo_id="revi13/ip-adapter-faceid-private",
filename="ip-adapter-faceid-plusv2_sd15.bin",
local_dir="/tmp/hf/models", # 保存先を手動指定
local_dir_use_symlinks=False, # シンボリックリンクではなく実体保存
token=os.environ["HUGGINGFACE_HUB_TOKEN"]
)
return {"model_path": model_path}
|