Skip to content

Commit e58fafe

Browse files
committed
Use Omit to define device config to prevent duplication
1 parent 3b115bf commit e58fafe

File tree

1 file changed

+1
-7
lines changed

1 file changed

+1
-7
lines changed

src/configs.js

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -415,11 +415,5 @@ export class AutoConfig {
415415

416416
/**
417417
* Device-specific configuration options.
418-
* @typedef {Object} DeviceConfig
419-
* @property {import('./utils/tensor.js').DataType|Record<import('./utils/dtypes.js').DataType, import('./utils/tensor.js').DataType>} [kv_cache_dtype] The data type of the key-value cache.
420-
* @property {Record<string, number>} [free_dimension_overrides] Override the free dimensions of the model.
421-
* See https://onnxruntime.ai/docs/tutorials/web/env-flags-and-session-options.html#freedimensionoverrides
422-
* for more information.
423-
* @property {import('./utils/dtypes.js').DataType|Record<string, import('./utils/dtypes.js').DataType>} [dtype] The default data type to use for the model.
424-
* @property {import('./utils/hub.js').ExternalData|Record<string, import('./utils/hub.js').ExternalData>} [use_external_data_format=false] Whether to load the model using the external data format (used for models >= 2GB in size).
418+
* @typedef {Omit<TransformersJSConfig, "device" | "device_config">} DeviceConfig
425419
*/

0 commit comments

Comments
 (0)