Spaces:
Runtime error
Runtime error
lionelgarnier
commited on
Commit
·
c35aeb6
1
Parent(s):
f22117a
debug
Browse files
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 |
)
|