Skip to content

Commit cbfdc07

Browse files
committed
Fix .from_pretrained function type
1 parent 3017402 commit cbfdc07

File tree

1 file changed

+10
-12
lines changed

1 file changed

+10
-12
lines changed

src/models.js

Lines changed: 10 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -3538,7 +3538,7 @@ export class CLIPModel extends CLIPPreTrainedModel { }
35383538
* The text model from CLIP without any head or projection on top.
35393539
*/
35403540
export class CLIPTextModel extends CLIPPreTrainedModel {
3541-
/** @type {PreTrainedModel.from_pretrained} */
3541+
/** @type {typeof PreTrainedModel.from_pretrained} */
35423542
static async from_pretrained(pretrained_model_name_or_path, options = {}) {
35433543
// Update default model file name if not provided
35443544
options.model_file_name ??= 'text_model';
@@ -3573,7 +3573,7 @@ export class CLIPTextModel extends CLIPPreTrainedModel {
35733573
* ```
35743574
*/
35753575
export class CLIPTextModelWithProjection extends CLIPPreTrainedModel {
3576-
/** @type {PreTrainedModel.from_pretrained} */
3576+
/** @type {typeof PreTrainedModel.from_pretrained} */
35773577
static async from_pretrained(pretrained_model_name_or_path, options = {}) {
35783578
// Update default model file name if not provided
35793579
options.model_file_name ??= 'text_model';
@@ -3585,7 +3585,7 @@ export class CLIPTextModelWithProjection extends CLIPPreTrainedModel {
35853585
* The vision model from CLIP without any head or projection on top.
35863586
*/
35873587
export class CLIPVisionModel extends CLIPPreTrainedModel {
3588-
/** @type {PreTrainedModel.from_pretrained} */
3588+
/** @type {typeof PreTrainedModel.from_pretrained} */
35893589
static async from_pretrained(pretrained_model_name_or_path, options = {}) {
35903590
// Update default model file name if not provided
35913591
options.model_file_name ??= 'vision_model';
@@ -3620,7 +3620,7 @@ export class CLIPVisionModel extends CLIPPreTrainedModel {
36203620
* ```
36213621
*/
36223622
export class CLIPVisionModelWithProjection extends CLIPPreTrainedModel {
3623-
/** @type {PreTrainedModel.from_pretrained} */
3623+
/** @type {typeof PreTrainedModel.from_pretrained} */
36243624
static async from_pretrained(pretrained_model_name_or_path, options = {}) {
36253625
// Update default model file name if not provided
36263626
options.model_file_name ??= 'vision_model';
@@ -3706,8 +3706,7 @@ export class SiglipModel extends SiglipPreTrainedModel { }
37063706
* ```
37073707
*/
37083708
export class SiglipTextModel extends SiglipPreTrainedModel {
3709-
3710-
/** @type {PreTrainedModel.from_pretrained} */
3709+
/** @type {typeof PreTrainedModel.from_pretrained} */
37113710
static async from_pretrained(pretrained_model_name_or_path, options = {}) {
37123711
// Update default model file name if not provided
37133712
options.model_file_name ??= 'text_model';
@@ -3742,7 +3741,7 @@ export class SiglipTextModel extends SiglipPreTrainedModel {
37423741
* ```
37433742
*/
37443743
export class SiglipVisionModel extends CLIPPreTrainedModel {
3745-
/** @type {PreTrainedModel.from_pretrained} */
3744+
/** @type {typeof PreTrainedModel.from_pretrained} */
37463745
static async from_pretrained(pretrained_model_name_or_path, options = {}) {
37473746
// Update default model file name if not provided
37483747
options.model_file_name ??= 'vision_model';
@@ -3763,7 +3762,7 @@ export class JinaCLIPPreTrainedModel extends PreTrainedModel { }
37633762
export class JinaCLIPModel extends JinaCLIPPreTrainedModel { }
37643763

37653764
export class JinaCLIPTextModel extends JinaCLIPPreTrainedModel {
3766-
/** @type {PreTrainedModel.from_pretrained} */
3765+
/** @type {typeof PreTrainedModel.from_pretrained} */
37673766
static async from_pretrained(pretrained_model_name_or_path, options = {}) {
37683767
// Update default model file name if not provided
37693768
options.model_file_name ??= 'text_model';
@@ -3772,7 +3771,7 @@ export class JinaCLIPTextModel extends JinaCLIPPreTrainedModel {
37723771
}
37733772

37743773
export class JinaCLIPVisionModel extends JinaCLIPPreTrainedModel {
3775-
/** @type {PreTrainedModel.from_pretrained} */
3774+
/** @type {typeof PreTrainedModel.from_pretrained} */
37763775
static async from_pretrained(pretrained_model_name_or_path, options = {}) {
37773776
// Update default model file name if not provided
37783777
options.model_file_name ??= 'vision_model';
@@ -5999,8 +5998,7 @@ export class ClapModel extends ClapPreTrainedModel { }
59995998
* ```
60005999
*/
60016000
export class ClapTextModelWithProjection extends ClapPreTrainedModel {
6002-
6003-
/** @type {PreTrainedModel.from_pretrained} */
6001+
/** @type {typeof PreTrainedModel.from_pretrained} */
60046002
static async from_pretrained(pretrained_model_name_or_path, options = {}) {
60056003
// Update default model file name if not provided
60066004
options.model_file_name ??= 'text_model';
@@ -6035,7 +6033,7 @@ export class ClapTextModelWithProjection extends ClapPreTrainedModel {
60356033
* ```
60366034
*/
60376035
export class ClapAudioModelWithProjection extends ClapPreTrainedModel {
6038-
/** @type {PreTrainedModel.from_pretrained} */
6036+
/** @type {typeof PreTrainedModel.from_pretrained} */
60396037
static async from_pretrained(pretrained_model_name_or_path, options = {}) {
60406038
// Update default model file name if not provided
60416039
options.model_file_name ??= 'audio_model';

0 commit comments

Comments
 (0)