Also update when chat_template is undefined

huggingface · Mar 16, 2024 · 926fbf5 · 926fbf5
1 parent 1ec55b3
commit 926fbf5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/tokenizers.js b/src/tokenizers.js
@@ -3038,10 +3038,10 @@ export class PreTrainedTokenizer extends Callable {
                     `template names are ${Object.keys(template_dict).sort()}.`
                 )
             }
-        } else if (chat_template === null) {
+        } else {
             // These are the cases when the model has a single template
             // priority: `chat_template` argument > `tokenizer.chat_template` > `tokenizer.default_chat_template
-            chat_template = this.chat_template ?? this.default_chat_template;
+            chat_template ??= this.chat_template ?? this.default_chat_template;
         }
 
         // Compilation function uses a cache to avoid recompiling the same template