Skip to content

Commit dc41231

Browse files
committed
LanguageModel: Fix generate() options
1 parent a658e7e commit dc41231

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/LanguageModel/index.js

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ class LanguageModel extends EventEmitter {
4040
default:
4141
throw 'Unrecognized model ' + modelNameOrUrl + ', try e.g. TinyStories-15M';
4242
}
43-
} else if (typeof modelNameOrUrl === 'object') {
43+
} else if (typeof modelNameOrOptions === 'object') {
4444
this.options.modelUrl = (typeof modelNameOrOptions.modelUrl === 'string') ? modelNameOrOptions.modelUrl : this.options.modelUrl;
4545
this.options.tokenizerUrl = (typeof modelNameOrOptions.tokenizerUrl === 'string') ? modelNameOrOptions.tokenizerUrl : this.options.tokenizerUrl;
4646
}

0 commit comments

Comments
 (0)