Rooobert commited on
Commit
666813f
·
verified ·
1 Parent(s): adcdd13

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ from langchain_core.prompts import PromptTemplate
9
  from transformers import pipeline
10
 
11
  # 載入 Mistral 模型
12
- model_path = "nvidia/Mistral-NeMo-Minitron-8B-Base"
13
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
14
  dtype = torch.bfloat16
15
  print(f"使用設備: {device}")
 
9
  from transformers import pipeline
10
 
11
  # 載入 Mistral 模型
12
+ model_path = "nvidia/Mistral-NeMo-Minitron-8B-instruct"
13
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
14
  dtype = torch.bfloat16
15
  print(f"使用設備: {device}")