lionelgarnier commited on
Commit
c35aeb6
·
1 Parent(s): f22117a
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -99,7 +99,7 @@ def get_text_gen_pipeline():
99
  try:
100
  device = "cuda" if torch.cuda.is_available() else "cpu"
101
  tokenizer = AutoTokenizer.from_pretrained(
102
- "mistralai/Mistral-7B-Instruct-v0.3",
103
  "deepseek-ai/DeepSeek-R1-Distill-Llama-8B",
104
  use_fast=True
105
  )
 
99
  try:
100
  device = "cuda" if torch.cuda.is_available() else "cpu"
101
  tokenizer = AutoTokenizer.from_pretrained(
102
+ # "mistralai/Mistral-7B-Instruct-v0.3",
103
  "deepseek-ai/DeepSeek-R1-Distill-Llama-8B",
104
  use_fast=True
105
  )