cwadayi commited on
Commit
cb4b734
·
verified ·
1 Parent(s): e579be1

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +12 -8
config.py CHANGED
@@ -4,11 +4,15 @@ import tempfile
4
  from datetime import datetime
5
 
6
  # --- 環境設定 ---
7
- # 設定 Matplotlib Hugging Face 模型的快取目錄
 
 
 
 
 
 
 
8
  os.environ.setdefault("MPLCONFIGDIR", "/tmp/matplotlib")
9
- os.environ.setdefault("TRANSFORMERS_CACHE", "/tmp/huggingface")
10
- TRANSFORMERS_CACHE = os.environ["TRANSFORMERS_CACHE"]
11
- os.makedirs(TRANSFORMERS_CACHE, exist_ok=True)
12
 
13
  # --- LINE Bot 憑證 ---
14
  CHANNEL_ACCESS_TOKEN = os.getenv("CHANNEL_ACCESS_TOKEN")
@@ -32,10 +36,10 @@ os.makedirs(STATIC_DIR, exist_ok=True)
32
  CWA_ALARM_API = "https://app-2.cwa.gov.tw/api/v1/earthquake/alarm/list"
33
  USGS_API_BASE_URL = "https://earthquake.usgs.gov/fdsnws/event/1/query"
34
 
35
- # --- AI 模型設定 (已更新為 Gemma) ---
36
- HUGGING_FACE_TOKEN = os.getenv("HUGGING_FACE_TOKEN") # [新增] 用於讀取 HF Token
37
- LLM_MODEL = os.getenv("LLM_MODEL", "google/gemma-2b-it") # [修改] 更新預設模型
38
- LLM_MAX_NEW_TOKENS = int(os.getenv("LLM_MAX_NEW_TOKENS", "256")) # [修改] 增加 token 數量
39
  LLM_TOP_K = int(os.getenv("LLM_TOP_K", "50"))
40
  LLM_TEMPERATURE = float(os.getenv("LLM_TEMPERATURE", "0.7"))
41
 
 
4
  from datetime import datetime
5
 
6
  # --- 環境設定 ---
7
+ # [修正] 使用 HF_HOME 來統一設定 Hugging Face 所有檔案的儲存路徑
8
+ # 這會告訴 huggingface_hub (用於 login) 和 transformers (用於下載模型)
9
+ # 將所有檔案都存放在一個可寫入的 /tmp 目錄下,解決 Permission denied 問題。
10
+ HF_HOME_DIR = "/tmp/huggingface"
11
+ os.environ["HF_HOME"] = HF_HOME_DIR
12
+ os.makedirs(HF_HOME_DIR, exist_ok=True)
13
+
14
+ # 設定 Matplotlib 的快取目錄
15
  os.environ.setdefault("MPLCONFIGDIR", "/tmp/matplotlib")
 
 
 
16
 
17
  # --- LINE Bot 憑證 ---
18
  CHANNEL_ACCESS_TOKEN = os.getenv("CHANNEL_ACCESS_TOKEN")
 
36
  CWA_ALARM_API = "https://app-2.cwa.gov.tw/api/v1/earthquake/alarm/list"
37
  USGS_API_BASE_URL = "https://earthquake.usgs.gov/fdsnws/event/1/query"
38
 
39
+ # --- AI 模型設定 (Gemma) ---
40
+ HUGGING_FACE_TOKEN = os.getenv("HUGGING_FACE_TOKEN")
41
+ LLM_MODEL = os.getenv("LLM_MODEL", "google/gemma-2b-it")
42
+ LLM_MAX_NEW_TOKENS = int(os.getenv("LLM_MAX_NEW_TOKENS", "256"))
43
  LLM_TOP_K = int(os.getenv("LLM_TOP_K", "50"))
44
  LLM_TEMPERATURE = float(os.getenv("LLM_TEMPERATURE", "0.7"))
45