Skip to content

Commit 92ae238

Browse files
authored
Expose more typescript exports to developers (#1325)
* Expose commonly-used types * Also expose device type
1 parent ff2d353 commit 92ae238

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

src/transformers.js

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,3 +40,11 @@ export * from './models/auto/processing_auto.js';
4040
export * from './generation/streamers.js';
4141
export * from './generation/stopping_criteria.js';
4242
export * from './generation/logits_process.js';
43+
44+
// Expose common types used across the library for developers to access
45+
/**
46+
* @typedef {import('./utils/hub.js').PretrainedModelOptions} PretrainedModelOptions
47+
* @typedef {import('./base/processing_utils.js').PretrainedProcessorOptions} PretrainedProcessorOptions
48+
* @typedef {import('./utils/dtypes.js').DataType} DataType
49+
* @typedef {import('./utils/devices.js').DeviceType} DeviceType
50+
*/

0 commit comments

Comments
 (0)