jordigonzm commited on
Commit
2728c6d
1 Parent(s): cee3ff2

Verificar y asignar pad_token_id si es None

Browse files
Files changed (1) hide show
  1. app.py +4 -0
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