Improve unknown model warnings (#554)

Warn with model type if known
huggingface · Feb 5, 2024 · a6f3f88 · a6f3f88
1 parent 2b3dcb1
commit a6f3f88
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/models.js b/src/models.js
@@ -788,7 +788,7 @@ export class PreTrainedModel extends Callable {
 
         } else { // should be MODEL_TYPES.EncoderOnly
             if (modelType !== MODEL_TYPES.EncoderOnly) {
-                console.warn(`Model type for '${modelName}' not found, assuming encoder-only architecture. Please report this at https://github.com/xenova/transformers.js/issues/new/choose.`)
+                console.warn(`Model type for '${modelName ?? config?.model_type}' not found, assuming encoder-only architecture. Please report this at https://github.com/xenova/transformers.js/issues/new/choose.`)
             }
             info = await Promise.all([
                 AutoConfig.from_pretrained(pretrained_model_name_or_path, options),