Skip to content

Commit 1eaee58

Browse files
committed
Also expose device type
1 parent e63c45f commit 1eaee58

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/transformers.js

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -41,8 +41,10 @@ export * from './generation/streamers.js';
4141
export * from './generation/stopping_criteria.js';
4242
export * from './generation/logits_process.js';
4343

44+
// Expose common types used across the library for developers to access
4445
/**
4546
* @typedef {import('./utils/hub.js').PretrainedModelOptions} PretrainedModelOptions
4647
* @typedef {import('./base/processing_utils.js').PretrainedProcessorOptions} PretrainedProcessorOptions
4748
* @typedef {import('./utils/dtypes.js').DataType} DataType
49+
* @typedef {import('./utils/devices.js').DeviceType} DeviceType
4850
*/

0 commit comments

Comments
 (0)