DMID23 commited on
Commit
6cd409f
·
verified ·
1 Parent(s): 0d6a1c8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -10,8 +10,8 @@ login(token)
10
 
11
  base_model = "mistralai/Mistral-7B-Instruct-v0.1"
12
 
13
- tokenizer = AutoTokenizer.from_pretrained(base_model, use_auth_token=True)
14
- model = AutoModelForCausalLM.from_pretrained(base_model, torch_dtype="auto", use_auth_token=True)
15
 
16
  # ---------- STEP 1: Fine-tuned 모델 Git에서 clone ----------
17
  repo_url = "DMID23/MachineToolAgent"
 
10
 
11
  base_model = "mistralai/Mistral-7B-Instruct-v0.1"
12
 
13
+ tokenizer = AutoTokenizer.from_pretrained(base_model, token=True)
14
+ model = AutoModelForCausalLM.from_pretrained(base_model, torch_dtype="auto", token=True)
15
 
16
  # ---------- STEP 1: Fine-tuned 모델 Git에서 clone ----------
17
  repo_url = "DMID23/MachineToolAgent"