Spaces:
Sleeping
Sleeping
jordigonzm
commited on
Commit
•
2728c6d
1
Parent(s):
cee3ff2
Verificar y asignar pad_token_id si es None
Browse files
app.py
CHANGED
@@ -70,6 +70,10 @@ def stream_chat(message: str, history: list, temperature: float, max_new_tokens:
|
|
70 |
# Limpieza de la historia para evitar pares con 'None'
|
71 |
cleaned_history = [[prompt, answer if answer is not None else ""] for prompt, answer in history]
|
72 |
|
|
|
|
|
|
|
|
|
73 |
stop = StopOnTokens()
|
74 |
|
75 |
# Preparar los input_ids y manejar la máscara de atención
|
|
|
70 |
# Limpieza de la historia para evitar pares con 'None'
|
71 |
cleaned_history = [[prompt, answer if answer is not None else ""] for prompt, answer in history]
|
72 |
|
73 |
+
# Verificar y asignar pad_token_id si es None
|
74 |
+
if tokenizer.pad_token_id is None:
|
75 |
+
tokenizer.pad_token_id = tokenizer.eos_token_id
|
76 |
+
|
77 |
stop = StopOnTokens()
|
78 |
|
79 |
# Preparar los input_ids y manejar la máscara de atención
|