transformers.js
e5994709 - [v4] Refactor models.js into separate per-model modeling files (#1498)

Commit
102 days ago
[v4] Refactor models.js into separate per-model modeling files (#1498) * started refactoring * started refactoring * started refactoring * added model class files * added model class files * added model class files * all model classes in their own files * refactored PreTrainedModel * refactoring done, lets fix bugs * added model-registry * removed dev file * changed casing * refactored MODEL_TYPE_CONFIG * fixed tests * small refactoring * moved model loader to its own file * fixed ts errors * big structure refactoring * fixed build * renamed _base/pre-trained-model.js and _base/output.js * small casing changes * Update src/models/ernie4_5/modeling_ernie4_5.js Co-authored-by: Joshua Lochner <admin@xenova.com> * refactored models/utils.js * fixed double MODEL_FOR_ definitions with registerTaskMappings helper * auto/image_processing_auto.js export * auto/image_processing_auto.js export * Improve model mapping setup * Fix LlavaPreTrainedModel * Move llava_onevision to separate files * Add missing exports * Update jinja version * Fix default class mapping * Simplify registerTaskMappings * Update registry.js * Formatting in src/models * Formatting in src * Move model-specific ModelOutput to respective modeling files * Final cleanup * Cleanup model exports * Fix Tensor type re-export * Clean up registry exports * Cleanup * Simplify loadResourceFile * Use positional arguments for repo id and filename * Update global library exports * Remove ts-expect-error * Formatting * let -> const --------- Co-authored-by: Joshua Lochner <admin@xenova.com> Co-authored-by: Joshua Lochner <26504141+xenova@users.noreply.github.com>
Author
Parents
Loading