NV9523 commited on
Commit
7b54bee
·
verified ·
1 Parent(s): 76a3b35

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -13,17 +13,16 @@ if HF_TOKEN:
13
  login(token=HF_TOKEN)
14
  print("✅ Đăng nhập thành công!")
15
  else:
16
- print("⚠️ Không tìm thấy HF_KEY trong Secrets!")
17
 
18
  # 🚀 Load model với key
19
  model_id = "meta-llama/Llama-3.2-1B-Instruct"
20
 
21
  print("⏳ Đang tải tokenizer & model...")
22
- tokenizer = AutoTokenizer.from_pretrained(model_id, token=HF_TOKEN, use_auth_token=HF_TOKEN)
23
  model = AutoModelForCausalLM.from_pretrained(
24
  model_id,
25
  token=HF_TOKEN,
26
- use_auth_token=HF_TOKEN,
27
  torch_dtype=torch.float32,
28
  low_cpu_mem_usage=True,
29
  device_map="cpu"
 
13
  login(token=HF_TOKEN)
14
  print("✅ Đăng nhập thành công!")
15
  else:
16
+ raise ValueError("⚠️ Không tìm thấy HF_Key trong Secrets!")
17
 
18
  # 🚀 Load model với key
19
  model_id = "meta-llama/Llama-3.2-1B-Instruct"
20
 
21
  print("⏳ Đang tải tokenizer & model...")
22
+ tokenizer = AutoTokenizer.from_pretrained(model_id, token=HF_TOKEN)
23
  model = AutoModelForCausalLM.from_pretrained(
24
  model_id,
25
  token=HF_TOKEN,
 
26
  torch_dtype=torch.float32,
27
  low_cpu_mem_usage=True,
28
  device_map="cpu"