Update funpack.py

2026-01-23 16:34:56 +02:00
parent 0261e1a7b3
commit e306d6e643
1 changed files with 5 additions and 0 deletions
@@ -538,6 +538,10 @@ class FunPackPromptEnhancer:
                {"role": "user", "content": user_prompt}
            ]

+            if llm_tokenizer.pad_token_id is None:
+                llm_tokenizer.pad_token = llm_tokenizer.eos_token
+                llm_tokenizer.pad_token_id = llm_tokenizer.eos_token_id
+
            llm_tokens = llm_tokenizer.apply_chat_template(
                messages,
                add_generation_prompt=True,
@@ -1186,5 +1190,6 @@ NODE_DISPLAY_NAME_MAPPINGS = {



+