Fix error when dtype in config is unset (#1271)

hans00 · web-flow · commit 991b2592755f · 2025-04-09T15:08:38.000-04:00
* Fix auto dtype

* Fix typecheck
diff --git a/src/models.js b/src/models.js
@@ -191,7 +191,7 @@ async function getSession(pretrained_model_name_or_path, fileName, options) {
         // Try to choose the auto dtype based on the custom config
         let config_dtype = custom_config.dtype;
         if (typeof config_dtype !== 'string') {
-            config_dtype = config_dtype[fileName];
+            config_dtype = config_dtype?.[fileName];
         }
 
         if (config_dtype && config_dtype !== DATA_TYPES.auto && DATA_TYPES.hasOwnProperty(config_dtype)) {

Original file line number	Diff line number	Diff line change
`@@ -191,7 +191,7 @@ async function getSession(pretrained_model_name_or_path, fileName, options) {`
`191`	`191`	`// Try to choose the auto dtype based on the custom config`
`192`	`192`	`let config_dtype = custom_config.dtype;`
`193`	`193`	`if (typeof config_dtype !== 'string') {`
`194`		`- config_dtype = config_dtype[fileName];`
	`194`	`+ config_dtype = config_dtype?.[fileName];`
`195`	`195`	`}`
`196`	`196`
`197`	`197`	`if (config_dtype && config_dtype !== DATA_TYPES.auto && DATA_TYPES.hasOwnProperty(config_dtype)) {`