DMID23 commited on
Commit
b657104
·
verified ·
1 Parent(s): bac07e4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -14,8 +14,8 @@ repo_url = "DMID23/MachineToolAgent"
14
 
15
 
16
  # ---------- STEP 2: Tokenizer와 모델 로드 ----------
17
- base_model = "mistralai/Mistral-7B-v0.1"
18
- tokenizer = AutoTokenizer.from_pretrained(base_model)
19
  model = AutoModelForCausalLM.from_pretrained(
20
  repo_url,
21
  torch_dtype=torch.float32,
 
14
 
15
 
16
  # ---------- STEP 2: Tokenizer와 모델 로드 ----------
17
+ #base_model = "mistralai/Mistral-7B-v0.1"
18
+ #tokenizer = AutoTokenizer.from_pretrained(base_model)
19
  model = AutoModelForCausalLM.from_pretrained(
20
  repo_url,
21
  torch_dtype=torch.float32,