@huggingface/tasks 0.21.25 → 0.21.27
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/commonjs/gguf.d.ts +3 -1
- package/dist/commonjs/gguf.d.ts.map +1 -1
- package/dist/commonjs/gguf.js +3 -0
- package/dist/commonjs/model-libraries-snippets.js +12 -12
- package/dist/commonjs/model-libraries.d.ts +8 -1
- package/dist/commonjs/model-libraries.d.ts.map +1 -1
- package/dist/commonjs/model-libraries.js +7 -0
- package/dist/commonjs/snippets/inputs.d.ts.map +1 -1
- package/dist/commonjs/snippets/inputs.js +4 -1
- package/dist/esm/gguf.d.ts +3 -1
- package/dist/esm/gguf.d.ts.map +1 -1
- package/dist/esm/gguf.js +3 -0
- package/dist/esm/model-libraries-snippets.js +12 -12
- package/dist/esm/model-libraries.d.ts +8 -1
- package/dist/esm/model-libraries.d.ts.map +1 -1
- package/dist/esm/model-libraries.js +7 -0
- package/dist/esm/snippets/inputs.d.ts.map +1 -1
- package/dist/esm/snippets/inputs.js +4 -1
- package/package.json +1 -1
- package/src/gguf.ts +3 -0
- package/src/model-libraries-snippets.ts +12 -12
- package/src/model-libraries.ts +7 -0
- package/src/snippets/inputs.ts +5 -1
package/dist/commonjs/gguf.d.ts
CHANGED
|
@@ -40,6 +40,7 @@ export declare enum GGMLFileQuantizationType {
|
|
|
40
40
|
MXFP4_MOE = 38,
|
|
41
41
|
NVFP4 = 39,
|
|
42
42
|
Q1_0 = 40,
|
|
43
|
+
Q2_0 = 41,
|
|
43
44
|
Q2_K_XL = 1000,
|
|
44
45
|
Q3_K_XL = 1001,
|
|
45
46
|
Q4_K_XL = 1002,
|
|
@@ -86,6 +87,7 @@ export declare enum GGMLQuantizationType {
|
|
|
86
87
|
TQ2_0 = 35,
|
|
87
88
|
MXFP4 = 39,
|
|
88
89
|
NVFP4 = 40,
|
|
89
|
-
Q1_0 = 41
|
|
90
|
+
Q1_0 = 41,
|
|
91
|
+
Q2_0 = 42
|
|
90
92
|
}
|
|
91
93
|
//# sourceMappingURL=gguf.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"gguf.d.ts","sourceRoot":"","sources":["../../src/gguf.ts"],"names":[],"mappings":"AAGA,oBAAY,wBAAwB;IACnC,GAAG,IAAI;IACP,GAAG,IAAI;IACP,IAAI,IAAI;IACR,IAAI,IAAI;IACR,aAAa,IAAI;IACjB,IAAI,IAAI;IACR,IAAI,IAAI;IACR,IAAI,IAAI;IACR,IAAI,IAAI;IACR,IAAI,IAAI;IACR,IAAI,KAAK;IACT,MAAM,KAAK;IACX,MAAM,KAAK;IACX,MAAM,KAAK;IACX,MAAM,KAAK;IACX,MAAM,KAAK;IACX,MAAM,KAAK;IACX,MAAM,KAAK;IACX,IAAI,KAAK;IACT,OAAO,KAAK;IACZ,MAAM,KAAK;IACX,MAAM,KAAK;IACX,MAAM,KAAK;IACX,OAAO,KAAK;IACZ,KAAK,KAAK;IACV,MAAM,KAAK;IACX,KAAK,KAAK;IACV,KAAK,KAAK;IACV,KAAK,KAAK;IACV,KAAK,KAAK;IACV,MAAM,KAAK;IACX,KAAK,KAAK;IACV,IAAI,KAAK;IACT,QAAQ,KAAK;IACb,QAAQ,KAAK;IACb,QAAQ,KAAK;IACb,KAAK,KAAK;IACV,KAAK,KAAK;IACV,SAAS,KAAK;IACd,KAAK,KAAK;IACV,IAAI,KAAK;IAIT,OAAO,OAAO;IACd,OAAO,OAAO;IACd,OAAO,OAAO;IACd,OAAO,OAAO;IACd,OAAO,OAAO;IACd,OAAO,OAAO;CACd;AAGD,eAAO,MAAM,aAAa,QAEzB,CAAC;AACF,eAAO,MAAM,oBAAoB,QAAiC,CAAC;AAEnE,wBAAgB,mBAAmB,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS,CAGrE;AAKD,eAAO,MAAM,gBAAgB,EAAE,wBAAwB,
|
|
1
|
+
{"version":3,"file":"gguf.d.ts","sourceRoot":"","sources":["../../src/gguf.ts"],"names":[],"mappings":"AAGA,oBAAY,wBAAwB;IACnC,GAAG,IAAI;IACP,GAAG,IAAI;IACP,IAAI,IAAI;IACR,IAAI,IAAI;IACR,aAAa,IAAI;IACjB,IAAI,IAAI;IACR,IAAI,IAAI;IACR,IAAI,IAAI;IACR,IAAI,IAAI;IACR,IAAI,IAAI;IACR,IAAI,KAAK;IACT,MAAM,KAAK;IACX,MAAM,KAAK;IACX,MAAM,KAAK;IACX,MAAM,KAAK;IACX,MAAM,KAAK;IACX,MAAM,KAAK;IACX,MAAM,KAAK;IACX,IAAI,KAAK;IACT,OAAO,KAAK;IACZ,MAAM,KAAK;IACX,MAAM,KAAK;IACX,MAAM,KAAK;IACX,OAAO,KAAK;IACZ,KAAK,KAAK;IACV,MAAM,KAAK;IACX,KAAK,KAAK;IACV,KAAK,KAAK;IACV,KAAK,KAAK;IACV,KAAK,KAAK;IACV,MAAM,KAAK;IACX,KAAK,KAAK;IACV,IAAI,KAAK;IACT,QAAQ,KAAK;IACb,QAAQ,KAAK;IACb,QAAQ,KAAK;IACb,KAAK,KAAK;IACV,KAAK,KAAK;IACV,SAAS,KAAK;IACd,KAAK,KAAK;IACV,IAAI,KAAK;IACT,IAAI,KAAK;IAIT,OAAO,OAAO;IACd,OAAO,OAAO;IACd,OAAO,OAAO;IACd,OAAO,OAAO;IACd,OAAO,OAAO;IACd,OAAO,OAAO;CACd;AAGD,eAAO,MAAM,aAAa,QAEzB,CAAC;AACF,eAAO,MAAM,oBAAoB,QAAiC,CAAC;AAEnE,wBAAgB,mBAAmB,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS,CAGrE;AAKD,eAAO,MAAM,gBAAgB,EAAE,wBAAwB,EA6DtD,CAAC;AAIF,wBAAgB,oBAAoB,CACnC,KAAK,EAAE,wBAAwB,EAC/B,eAAe,EAAE,wBAAwB,EAAE,GACzC,wBAAwB,GAAG,SAAS,CAmCtC;AAGD,oBAAY,oBAAoB;IAC/B,GAAG,IAAI;IACP,GAAG,IAAI;IACP,IAAI,IAAI;IACR,IAAI,IAAI;IACR,IAAI,IAAI;IACR,IAAI,IAAI;IACR,IAAI,IAAI;IACR,IAAI,IAAI;IACR,IAAI,KAAK;IACT,IAAI,KAAK;IACT,IAAI,KAAK;IACT,IAAI,KAAK;IACT,IAAI,KAAK;IACT,IAAI,KAAK;IACT,OAAO,KAAK;IACZ,MAAM,KAAK;IACX,OAAO,KAAK;IACZ,KAAK,KAAK;IACV,MAAM,KAAK;IACX,KAAK,KAAK;IACV,KAAK,KAAK;IACV,MAAM,KAAK;IACX,EAAE,KAAK;IACP,GAAG,KAAK;IACR,GAAG,KAAK;IACR,GAAG,KAAK;IACR,GAAG,KAAK;IACR,KAAK,KAAK;IACV,IAAI,KAAK;IACT,KAAK,KAAK;IACV,KAAK,KAAK;IACV,KAAK,KAAK;IACV,KAAK,KAAK;IACV,IAAI,KAAK;IACT,IAAI,KAAK;CACT"}
|
package/dist/commonjs/gguf.js
CHANGED
|
@@ -49,6 +49,7 @@ var GGMLFileQuantizationType;
|
|
|
49
49
|
GGMLFileQuantizationType[GGMLFileQuantizationType["MXFP4_MOE"] = 38] = "MXFP4_MOE";
|
|
50
50
|
GGMLFileQuantizationType[GGMLFileQuantizationType["NVFP4"] = 39] = "NVFP4";
|
|
51
51
|
GGMLFileQuantizationType[GGMLFileQuantizationType["Q1_0"] = 40] = "Q1_0";
|
|
52
|
+
GGMLFileQuantizationType[GGMLFileQuantizationType["Q2_0"] = 41] = "Q2_0";
|
|
52
53
|
// custom quants used by unsloth
|
|
53
54
|
// they are not officially a scheme enum value in GGUF, but only here for naming
|
|
54
55
|
GGMLFileQuantizationType[GGMLFileQuantizationType["Q2_K_XL"] = 1000] = "Q2_K_XL";
|
|
@@ -116,6 +117,7 @@ exports.GGUF_QUANT_ORDER = [
|
|
|
116
117
|
GGMLFileQuantizationType.IQ2_S,
|
|
117
118
|
GGMLFileQuantizationType.IQ2_XS,
|
|
118
119
|
GGMLFileQuantizationType.IQ2_XXS,
|
|
120
|
+
GGMLFileQuantizationType.Q2_0,
|
|
119
121
|
// 1-bit quantizations
|
|
120
122
|
GGMLFileQuantizationType.IQ1_S,
|
|
121
123
|
GGMLFileQuantizationType.IQ1_M,
|
|
@@ -193,4 +195,5 @@ var GGMLQuantizationType;
|
|
|
193
195
|
GGMLQuantizationType[GGMLQuantizationType["MXFP4"] = 39] = "MXFP4";
|
|
194
196
|
GGMLQuantizationType[GGMLQuantizationType["NVFP4"] = 40] = "NVFP4";
|
|
195
197
|
GGMLQuantizationType[GGMLQuantizationType["Q1_0"] = 41] = "Q1_0";
|
|
198
|
+
GGMLQuantizationType[GGMLQuantizationType["Q2_0"] = 42] = "Q2_0";
|
|
196
199
|
})(GGMLQuantizationType || (exports.GGMLQuantizationType = GGMLQuantizationType = {}));
|
|
@@ -384,7 +384,7 @@ const diffusers_default = (model) => [
|
|
|
384
384
|
from diffusers import DiffusionPipeline
|
|
385
385
|
|
|
386
386
|
# switch to "mps" for apple devices
|
|
387
|
-
pipe = DiffusionPipeline.from_pretrained("${model.id}",
|
|
387
|
+
pipe = DiffusionPipeline.from_pretrained("${model.id}", torch_dtype=torch.bfloat16, device_map="cuda")
|
|
388
388
|
|
|
389
389
|
prompt = "${get_prompt_from_diffusers_model(model) ?? diffusersDefaultPrompt}"
|
|
390
390
|
image = pipe(prompt).images[0]`,
|
|
@@ -395,7 +395,7 @@ from diffusers import DiffusionPipeline
|
|
|
395
395
|
from diffusers.utils import load_image
|
|
396
396
|
|
|
397
397
|
# switch to "mps" for apple devices
|
|
398
|
-
pipe = DiffusionPipeline.from_pretrained("${model.id}",
|
|
398
|
+
pipe = DiffusionPipeline.from_pretrained("${model.id}", torch_dtype=torch.bfloat16, device_map="cuda")
|
|
399
399
|
|
|
400
400
|
prompt = "${get_prompt_from_diffusers_model(model) ?? diffusersImg2ImgDefaultPrompt}"
|
|
401
401
|
input_image = load_image("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/cat.png")
|
|
@@ -408,7 +408,7 @@ from diffusers import DiffusionPipeline
|
|
|
408
408
|
from diffusers.utils import load_image, export_to_video
|
|
409
409
|
|
|
410
410
|
# switch to "mps" for apple devices
|
|
411
|
-
pipe = DiffusionPipeline.from_pretrained("${model.id}",
|
|
411
|
+
pipe = DiffusionPipeline.from_pretrained("${model.id}", torch_dtype=torch.bfloat16, device_map="cuda")
|
|
412
412
|
pipe.to("cuda")
|
|
413
413
|
|
|
414
414
|
prompt = "${get_prompt_from_diffusers_model(model) ?? diffusersVideoDefaultPrompt}"
|
|
@@ -432,7 +432,7 @@ const diffusers_lora = (model) => [
|
|
|
432
432
|
from diffusers import DiffusionPipeline
|
|
433
433
|
|
|
434
434
|
# switch to "mps" for apple devices
|
|
435
|
-
pipe = DiffusionPipeline.from_pretrained("${get_base_diffusers_model(model)}",
|
|
435
|
+
pipe = DiffusionPipeline.from_pretrained("${get_base_diffusers_model(model)}", torch_dtype=torch.bfloat16, device_map="cuda")
|
|
436
436
|
pipe.load_lora_weights("${model.id}")
|
|
437
437
|
|
|
438
438
|
prompt = "${get_prompt_from_diffusers_model(model) ?? diffusersDefaultPrompt}"
|
|
@@ -444,7 +444,7 @@ from diffusers import DiffusionPipeline
|
|
|
444
444
|
from diffusers.utils import load_image
|
|
445
445
|
|
|
446
446
|
# switch to "mps" for apple devices
|
|
447
|
-
pipe = DiffusionPipeline.from_pretrained("${get_base_diffusers_model(model)}",
|
|
447
|
+
pipe = DiffusionPipeline.from_pretrained("${get_base_diffusers_model(model)}", torch_dtype=torch.bfloat16, device_map="cuda")
|
|
448
448
|
pipe.load_lora_weights("${model.id}")
|
|
449
449
|
|
|
450
450
|
prompt = "${get_prompt_from_diffusers_model(model) ?? diffusersImg2ImgDefaultPrompt}"
|
|
@@ -458,7 +458,7 @@ from diffusers import DiffusionPipeline
|
|
|
458
458
|
from diffusers.utils import export_to_video
|
|
459
459
|
|
|
460
460
|
# switch to "mps" for apple devices
|
|
461
|
-
pipe = DiffusionPipeline.from_pretrained("${get_base_diffusers_model(model)}",
|
|
461
|
+
pipe = DiffusionPipeline.from_pretrained("${get_base_diffusers_model(model)}", torch_dtype=torch.bfloat16, device_map="cuda")
|
|
462
462
|
pipe.load_lora_weights("${model.id}")
|
|
463
463
|
|
|
464
464
|
prompt = "${get_prompt_from_diffusers_model(model) ?? diffusersVideoDefaultPrompt}"
|
|
@@ -472,7 +472,7 @@ from diffusers import DiffusionPipeline
|
|
|
472
472
|
from diffusers.utils import load_image, export_to_video
|
|
473
473
|
|
|
474
474
|
# switch to "mps" for apple devices
|
|
475
|
-
pipe = DiffusionPipeline.from_pretrained("${get_base_diffusers_model(model)}",
|
|
475
|
+
pipe = DiffusionPipeline.from_pretrained("${get_base_diffusers_model(model)}", torch_dtype=torch.bfloat16, device_map="cuda")
|
|
476
476
|
pipe.load_lora_weights("${model.id}")
|
|
477
477
|
|
|
478
478
|
prompt = "${get_prompt_from_diffusers_model(model) ?? diffusersVideoDefaultPrompt}"
|
|
@@ -486,7 +486,7 @@ const diffusers_textual_inversion = (model) => [
|
|
|
486
486
|
from diffusers import DiffusionPipeline
|
|
487
487
|
|
|
488
488
|
# switch to "mps" for apple devices
|
|
489
|
-
pipe = DiffusionPipeline.from_pretrained("${get_base_diffusers_model(model)}",
|
|
489
|
+
pipe = DiffusionPipeline.from_pretrained("${get_base_diffusers_model(model)}", torch_dtype=torch.bfloat16, device_map="cuda")
|
|
490
490
|
pipe.load_textual_inversion("${model.id}")`,
|
|
491
491
|
];
|
|
492
492
|
const diffusers_flux_fill = (model) => [
|
|
@@ -498,7 +498,7 @@ image = load_image("https://huggingface.co/datasets/diffusers/diffusers-images-d
|
|
|
498
498
|
mask = load_image("https://huggingface.co/datasets/diffusers/diffusers-images-docs/resolve/main/cup_mask.png")
|
|
499
499
|
|
|
500
500
|
# switch to "mps" for apple devices
|
|
501
|
-
pipe = FluxFillPipeline.from_pretrained("${model.id}",
|
|
501
|
+
pipe = FluxFillPipeline.from_pretrained("${model.id}", torch_dtype=torch.bfloat16, device_map="cuda")
|
|
502
502
|
image = pipe(
|
|
503
503
|
prompt="a white paper cup",
|
|
504
504
|
image=image,
|
|
@@ -518,7 +518,7 @@ from diffusers import AutoPipelineForInpainting
|
|
|
518
518
|
from diffusers.utils import load_image
|
|
519
519
|
|
|
520
520
|
# switch to "mps" for apple devices
|
|
521
|
-
pipe = AutoPipelineForInpainting.from_pretrained("${model.id}",
|
|
521
|
+
pipe = AutoPipelineForInpainting.from_pretrained("${model.id}", torch_dtype=torch.float16, variant="fp16", device_map="cuda")
|
|
522
522
|
|
|
523
523
|
img_url = "https://raw.githubusercontent.com/CompVis/latent-diffusion/main/data/inpainting_examples/overture-creations-5sI6fQgYIuo.png"
|
|
524
524
|
mask_url = "https://raw.githubusercontent.com/CompVis/latent-diffusion/main/data/inpainting_examples/overture-creations-5sI6fQgYIuo_mask.png"
|
|
@@ -1756,7 +1756,7 @@ const transformers = (model) => {
|
|
|
1756
1756
|
: info.processor === "AutoFeatureExtractor"
|
|
1757
1757
|
? "extractor"
|
|
1758
1758
|
: "processor";
|
|
1759
|
-
autoSnippet.push("# Load model directly", `from transformers import ${info.processor}, ${info.auto_model}`, "", `${processorVarName} = ${info.processor}.from_pretrained("${model.id}"` + remote_code_snippet + ")", `model = ${info.auto_model}.from_pretrained("${model.id}"` + remote_code_snippet + "
|
|
1759
|
+
autoSnippet.push("# Load model directly", `from transformers import ${info.processor}, ${info.auto_model}`, "", `${processorVarName} = ${info.processor}.from_pretrained("${model.id}"` + remote_code_snippet + ")", `model = ${info.auto_model}.from_pretrained("${model.id}"` + remote_code_snippet + ', device_map="auto")');
|
|
1760
1760
|
if (model.tags.includes("conversational") && hasChatTemplate(model)) {
|
|
1761
1761
|
if (model.tags.includes("image-text-to-text")) {
|
|
1762
1762
|
autoSnippet.push("messages = [", [
|
|
@@ -1776,7 +1776,7 @@ const transformers = (model) => {
|
|
|
1776
1776
|
}
|
|
1777
1777
|
}
|
|
1778
1778
|
else {
|
|
1779
|
-
autoSnippet.push("# Load model directly", `from transformers import ${info.auto_model}`, `model = ${info.auto_model}.from_pretrained("${model.id}"` + remote_code_snippet + ',
|
|
1779
|
+
autoSnippet.push("# Load model directly", `from transformers import ${info.auto_model}`, `model = ${info.auto_model}.from_pretrained("${model.id}"` + remote_code_snippet + ', device_map="auto")');
|
|
1780
1780
|
}
|
|
1781
1781
|
if (model.pipeline_tag && library_to_tasks_js_1.LIBRARY_TASK_MAPPING.transformers?.includes(model.pipeline_tag)) {
|
|
1782
1782
|
const pipelineSnippet = ["# Use a pipeline as a high-level helper"];
|
|
@@ -217,6 +217,13 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
|
|
|
217
217
|
repoUrl: string;
|
|
218
218
|
snippets: (model: ModelData) => string[];
|
|
219
219
|
};
|
|
220
|
+
ccpfn: {
|
|
221
|
+
prettyLabel: string;
|
|
222
|
+
repoName: string;
|
|
223
|
+
repoUrl: string;
|
|
224
|
+
filter: false;
|
|
225
|
+
countDownloads: string;
|
|
226
|
+
};
|
|
220
227
|
champ: {
|
|
221
228
|
prettyLabel: string;
|
|
222
229
|
repoName: string;
|
|
@@ -1739,5 +1746,5 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
|
|
|
1739
1746
|
};
|
|
1740
1747
|
export type ModelLibraryKey = keyof typeof MODEL_LIBRARIES_UI_ELEMENTS;
|
|
1741
1748
|
export declare const ALL_MODEL_LIBRARY_KEYS: ModelLibraryKey[];
|
|
1742
|
-
export declare const ALL_DISPLAY_MODEL_LIBRARY_KEYS: ("acestep" | "adapter-transformers" | "allennlp" | "anemoi" | "araclip" | "aviation-ner" | "asteroid" | "audiocraft" | "audioseal" | "bagel-mot" | "bboxmaskpose" | "ben2" | "bertopic" | "big_vision" | "bionemo" | "birder" | "birefnet" | "bm25s" | "boltzgen" | "cancertathomev2" | "cartesia_pytorch" | "cartesia_mlx" | "champ" | "chatterbox" | "chaossim" | "chat_tts" | "chexmix" | "chronos-forecasting" | "clara" | "clipscope" | "cloud-agents" | "collectorvision" | "colipri" | "cosyvoice" | "cotracker" | "colpali" | "comet" | "cosmos" | "cxr-foundation" | "deepforest" | "depth-anything-v2" | "depth-pro" | "derm-foundation" | "describe-anything" | "dia-tts" | "dia2" | "diff-interpretation-tuning" | "diffree" | "diffusers" | "diffusionkit" | "docking-at-home" | "doctr" | "edsnlp" | "elm" | "encoderfile" | "espnet" | "eupe" | "fairseq" | "fastai" | "fastprint" | "fasttext" | "fixer" | "flair" | "flextab" | "fme" | "gemma.cpp" | "geometry-crafter" | "gliner" | "gliner2" | "glm-tts" | "glyph-byt5" | "granite-library" | "grok" | "habibi-tts" | "hallo" | "hermes" | "holomotion" | "hezar" | "htrflow" | "hunyuan-dit" | "hunyuan3d-2" | "hunyuanworld-voyager" | "hy-worldplay" | "hy-world-2" | "image-matching-models" | "imstoucan" | "index-tts" | "infinitetalk" | "infinite-you" | "intellifold" | "ising-decoding" | "keras" | "tf-keras" | "keras-hub" | "kernels" | "kimi-audio" | "kittentts" | "kronos" | "k2" | "lyra-2.0" | "lagernvs" | "lightning-ir" | "litert" | "litert-lm" | "lerobot" | "lightglue" | "liveportrait" | "longcat-video-avatar-1.5" | "llama-cpp-python" | "mini-omni2" | "mindspore" | "magi-1" | "magenta-realtime" | "magenta-realtime-2" | "mamba-ssm" | "manas-1" | "mars5-tts" | "matanyone" | "mesh-anything" | "merlin" | "medvae" | "mitie" | "ml-agents" | "ml-sharp" | "mlx" | "mlx-image" | "mlc-llm" | "model2vec" | "mobilint" | "moshi" | "mtvcraft" | "multimolecule" | "nemo" | "nv-medtech" | "open-oasis" | "open_clip" | "openasr" | "openpeerllm" | "open-sora" | "outetts" | "paddlenlp" | "PaddleOCR" | "peft" | "perception-encoder" | "phantom-wan" | "pocket-tts" | "pruna-ai" | "pxia" | "pyannote-audio" | "py-feat" | "pythae" | "quantumpeer" | "qwen3_tts" | "recurrentgemma" | "relik" | "refiners" | "renderformer" | "reverb" | "rkllm" | "robo-orchard-lab" | "rwkv" | "saelens" | "scail-2" | "sam2" | "sam-3d-body" | "sam-3d-objects" | "same" | "sample-factory" | "sap-rpt-1-oss" | "sapiens" | "sapiens2" | "seedvr" | "self-forcing" | "sentence-transformers" | "setfit" | "sklearn" | "spacy" | "span-marker" | "speechbrain" | "ssr-speech" | "stable-audio-3" | "stable-audio-tools" | "monkeyocr" | "diffusion-single-file" | "seed-story" | "skala" | "soloaudio" | "songbloom" | "stable-baselines3" | "stanza" | "supertonic" | "swarmformer" | "synthefy-migas" | "f5-tts" | "genmo" | "tencent-song-generation" | "tensorflowtts" | "tensorrt" | "tabpfn" | "terratorch" | "tic-clip" | "timesfm" | "timee-ts" | "timm" | "tirex" | "tirex-2" | "torchgeo" | "transformers" | "transformers.js" | "trellis" | "trellis2" | "tunejury" | "ultralytics" | "univa" | "uni-3dar" | "unity-sentis" | "sana" | "videoprism" | "vfi-mamba" | "vismatch" | "lvface" | "voicecraft" | "voxcpm" | "vui" | "vibevoice" | "videox_fun" | "wan2.2" | "wham" | "whisperkit" | "yolov10" | "yolov26" | "zonos" | "3dtopia-xl")[];
|
|
1749
|
+
export declare const ALL_DISPLAY_MODEL_LIBRARY_KEYS: ("acestep" | "adapter-transformers" | "allennlp" | "anemoi" | "araclip" | "aviation-ner" | "asteroid" | "audiocraft" | "audioseal" | "bagel-mot" | "bboxmaskpose" | "ben2" | "bertopic" | "big_vision" | "bionemo" | "birder" | "birefnet" | "bm25s" | "boltzgen" | "cancertathomev2" | "cartesia_pytorch" | "cartesia_mlx" | "ccpfn" | "champ" | "chatterbox" | "chaossim" | "chat_tts" | "chexmix" | "chronos-forecasting" | "clara" | "clipscope" | "cloud-agents" | "collectorvision" | "colipri" | "cosyvoice" | "cotracker" | "colpali" | "comet" | "cosmos" | "cxr-foundation" | "deepforest" | "depth-anything-v2" | "depth-pro" | "derm-foundation" | "describe-anything" | "dia-tts" | "dia2" | "diff-interpretation-tuning" | "diffree" | "diffusers" | "diffusionkit" | "docking-at-home" | "doctr" | "edsnlp" | "elm" | "encoderfile" | "espnet" | "eupe" | "fairseq" | "fastai" | "fastprint" | "fasttext" | "fixer" | "flair" | "flextab" | "fme" | "gemma.cpp" | "geometry-crafter" | "gliner" | "gliner2" | "glm-tts" | "glyph-byt5" | "granite-library" | "grok" | "habibi-tts" | "hallo" | "hermes" | "holomotion" | "hezar" | "htrflow" | "hunyuan-dit" | "hunyuan3d-2" | "hunyuanworld-voyager" | "hy-worldplay" | "hy-world-2" | "image-matching-models" | "imstoucan" | "index-tts" | "infinitetalk" | "infinite-you" | "intellifold" | "ising-decoding" | "keras" | "tf-keras" | "keras-hub" | "kernels" | "kimi-audio" | "kittentts" | "kronos" | "k2" | "lyra-2.0" | "lagernvs" | "lightning-ir" | "litert" | "litert-lm" | "lerobot" | "lightglue" | "liveportrait" | "longcat-video-avatar-1.5" | "llama-cpp-python" | "mini-omni2" | "mindspore" | "magi-1" | "magenta-realtime" | "magenta-realtime-2" | "mamba-ssm" | "manas-1" | "mars5-tts" | "matanyone" | "mesh-anything" | "merlin" | "medvae" | "mitie" | "ml-agents" | "ml-sharp" | "mlx" | "mlx-image" | "mlc-llm" | "model2vec" | "mobilint" | "moshi" | "mtvcraft" | "multimolecule" | "nemo" | "nv-medtech" | "open-oasis" | "open_clip" | "openasr" | "openpeerllm" | "open-sora" | "outetts" | "paddlenlp" | "PaddleOCR" | "peft" | "perception-encoder" | "phantom-wan" | "pocket-tts" | "pruna-ai" | "pxia" | "pyannote-audio" | "py-feat" | "pythae" | "quantumpeer" | "qwen3_tts" | "recurrentgemma" | "relik" | "refiners" | "renderformer" | "reverb" | "rkllm" | "robo-orchard-lab" | "rwkv" | "saelens" | "scail-2" | "sam2" | "sam-3d-body" | "sam-3d-objects" | "same" | "sample-factory" | "sap-rpt-1-oss" | "sapiens" | "sapiens2" | "seedvr" | "self-forcing" | "sentence-transformers" | "setfit" | "sklearn" | "spacy" | "span-marker" | "speechbrain" | "ssr-speech" | "stable-audio-3" | "stable-audio-tools" | "monkeyocr" | "diffusion-single-file" | "seed-story" | "skala" | "soloaudio" | "songbloom" | "stable-baselines3" | "stanza" | "supertonic" | "swarmformer" | "synthefy-migas" | "f5-tts" | "genmo" | "tencent-song-generation" | "tensorflowtts" | "tensorrt" | "tabpfn" | "terratorch" | "tic-clip" | "timesfm" | "timee-ts" | "timm" | "tirex" | "tirex-2" | "torchgeo" | "transformers" | "transformers.js" | "trellis" | "trellis2" | "tunejury" | "ultralytics" | "univa" | "uni-3dar" | "unity-sentis" | "sana" | "videoprism" | "vfi-mamba" | "vismatch" | "lvface" | "voicecraft" | "voxcpm" | "vui" | "vibevoice" | "videox_fun" | "wan2.2" | "wham" | "whisperkit" | "yolov10" | "yolov26" | "zonos" | "3dtopia-xl")[];
|
|
1743
1750
|
//# sourceMappingURL=model-libraries.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"model-libraries.d.ts","sourceRoot":"","sources":["../../src/model-libraries.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AACjD,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,gCAAgC,CAAC;AAEzE;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAChC;;;;OAIG;IACH,WAAW,EAAE,MAAM,CAAC;IACpB;;OAEG;IACH,QAAQ,EAAE,MAAM,CAAC;IACjB;;OAEG;IACH,OAAO,EAAE,MAAM,CAAC;IAChB;;OAEG;IACH,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB;;OAEG;IACH,QAAQ,CAAC,EAAE,CAAC,KAAK,EAAE,SAAS,KAAK,MAAM,EAAE,CAAC;IAC1C;;;;;OAKG;IACH,cAAc,CAAC,EAAE,kBAAkB,CAAC;IACpC;;;OAGG;IACH,MAAM,CAAC,EAAE,OAAO,CAAC;CACjB;AAED;;;;;;;;;;;;;GAaG;AAEH,eAAO,MAAM,2BAA2B
|
|
1
|
+
{"version":3,"file":"model-libraries.d.ts","sourceRoot":"","sources":["../../src/model-libraries.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AACjD,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,gCAAgC,CAAC;AAEzE;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAChC;;;;OAIG;IACH,WAAW,EAAE,MAAM,CAAC;IACpB;;OAEG;IACH,QAAQ,EAAE,MAAM,CAAC;IACjB;;OAEG;IACH,OAAO,EAAE,MAAM,CAAC;IAChB;;OAEG;IACH,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB;;OAEG;IACH,QAAQ,CAAC,EAAE,CAAC,KAAK,EAAE,SAAS,KAAK,MAAM,EAAE,CAAC;IAC1C;;;;;OAKG;IACH,cAAc,CAAC,EAAE,kBAAkB,CAAC;IACpC;;;OAGG;IACH,MAAM,CAAC,EAAE,OAAO,CAAC;CACjB;AAED;;;;;;;;;;;;;GAaG;AAEH,eAAO,MAAM,2BAA2B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAiqDI,CAAC;AAE7C,MAAM,MAAM,eAAe,GAAG,MAAM,OAAO,2BAA2B,CAAC;AAEvE,eAAO,MAAM,sBAAsB,EAA+C,eAAe,EAAE,CAAC;AAEpG,eAAO,MAAM,8BAA8B,+vGAQ1B,CAAC"}
|
|
@@ -213,6 +213,13 @@ exports.MODEL_LIBRARIES_UI_ELEMENTS = {
|
|
|
213
213
|
repoUrl: "https://github.com/cartesia-ai/cartesia_mlx",
|
|
214
214
|
snippets: snippets.cartesia_mlx,
|
|
215
215
|
},
|
|
216
|
+
ccpfn: {
|
|
217
|
+
prettyLabel: "CCPFN",
|
|
218
|
+
repoName: "CCPFN",
|
|
219
|
+
repoUrl: "https://huggingface.co/Layer6/CCPFN",
|
|
220
|
+
filter: false,
|
|
221
|
+
countDownloads: `path_extension:"pt"`,
|
|
222
|
+
},
|
|
216
223
|
champ: {
|
|
217
224
|
prettyLabel: "Champ",
|
|
218
225
|
repoName: "Champ",
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"inputs.d.ts","sourceRoot":"","sources":["../../../src/snippets/inputs.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,0BAA0B,EAAE,MAAM,mBAAmB,CAAC;AACpE,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;
|
|
1
|
+
{"version":3,"file":"inputs.d.ts","sourceRoot":"","sources":["../../../src/snippets/inputs.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,0BAA0B,EAAE,MAAM,mBAAmB,CAAC;AACpE,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAyKnD,wBAAgB,oBAAoB,CACnC,KAAK,EAAE,gBAAgB,EACvB,MAAM,UAAQ,EACd,QAAQ,UAAQ,GACd,MAAM,GAAG,0BAA0B,EAAE,CAmBvC"}
|
|
@@ -51,7 +51,10 @@ const inputsTextGeneration = (model) => {
|
|
|
51
51
|
}
|
|
52
52
|
return `"Can you please let us know more details about your "`;
|
|
53
53
|
};
|
|
54
|
-
|
|
54
|
+
// `mask_token` is author-controlled (tokenizer/config.json); build the string then
|
|
55
|
+
// JSON.stringify it so quotes/backslashes are escaped into a valid string literal
|
|
56
|
+
// and cannot break out of the generated snippet.
|
|
57
|
+
const inputsFillMask = (model) => JSON.stringify(`The answer to the universe is ${model.mask_token}.`);
|
|
55
58
|
const inputsSentenceSimilarity = () => `{
|
|
56
59
|
"source_sentence": "That is a happy person",
|
|
57
60
|
"sentences": [
|
package/dist/esm/gguf.d.ts
CHANGED
|
@@ -40,6 +40,7 @@ export declare enum GGMLFileQuantizationType {
|
|
|
40
40
|
MXFP4_MOE = 38,
|
|
41
41
|
NVFP4 = 39,
|
|
42
42
|
Q1_0 = 40,
|
|
43
|
+
Q2_0 = 41,
|
|
43
44
|
Q2_K_XL = 1000,
|
|
44
45
|
Q3_K_XL = 1001,
|
|
45
46
|
Q4_K_XL = 1002,
|
|
@@ -86,6 +87,7 @@ export declare enum GGMLQuantizationType {
|
|
|
86
87
|
TQ2_0 = 35,
|
|
87
88
|
MXFP4 = 39,
|
|
88
89
|
NVFP4 = 40,
|
|
89
|
-
Q1_0 = 41
|
|
90
|
+
Q1_0 = 41,
|
|
91
|
+
Q2_0 = 42
|
|
90
92
|
}
|
|
91
93
|
//# sourceMappingURL=gguf.d.ts.map
|
package/dist/esm/gguf.d.ts.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"gguf.d.ts","sourceRoot":"","sources":["../../src/gguf.ts"],"names":[],"mappings":"AAGA,oBAAY,wBAAwB;IACnC,GAAG,IAAI;IACP,GAAG,IAAI;IACP,IAAI,IAAI;IACR,IAAI,IAAI;IACR,aAAa,IAAI;IACjB,IAAI,IAAI;IACR,IAAI,IAAI;IACR,IAAI,IAAI;IACR,IAAI,IAAI;IACR,IAAI,IAAI;IACR,IAAI,KAAK;IACT,MAAM,KAAK;IACX,MAAM,KAAK;IACX,MAAM,KAAK;IACX,MAAM,KAAK;IACX,MAAM,KAAK;IACX,MAAM,KAAK;IACX,MAAM,KAAK;IACX,IAAI,KAAK;IACT,OAAO,KAAK;IACZ,MAAM,KAAK;IACX,MAAM,KAAK;IACX,MAAM,KAAK;IACX,OAAO,KAAK;IACZ,KAAK,KAAK;IACV,MAAM,KAAK;IACX,KAAK,KAAK;IACV,KAAK,KAAK;IACV,KAAK,KAAK;IACV,KAAK,KAAK;IACV,MAAM,KAAK;IACX,KAAK,KAAK;IACV,IAAI,KAAK;IACT,QAAQ,KAAK;IACb,QAAQ,KAAK;IACb,QAAQ,KAAK;IACb,KAAK,KAAK;IACV,KAAK,KAAK;IACV,SAAS,KAAK;IACd,KAAK,KAAK;IACV,IAAI,KAAK;IAIT,OAAO,OAAO;IACd,OAAO,OAAO;IACd,OAAO,OAAO;IACd,OAAO,OAAO;IACd,OAAO,OAAO;IACd,OAAO,OAAO;CACd;AAGD,eAAO,MAAM,aAAa,QAEzB,CAAC;AACF,eAAO,MAAM,oBAAoB,QAAiC,CAAC;AAEnE,wBAAgB,mBAAmB,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS,CAGrE;AAKD,eAAO,MAAM,gBAAgB,EAAE,wBAAwB,
|
|
1
|
+
{"version":3,"file":"gguf.d.ts","sourceRoot":"","sources":["../../src/gguf.ts"],"names":[],"mappings":"AAGA,oBAAY,wBAAwB;IACnC,GAAG,IAAI;IACP,GAAG,IAAI;IACP,IAAI,IAAI;IACR,IAAI,IAAI;IACR,aAAa,IAAI;IACjB,IAAI,IAAI;IACR,IAAI,IAAI;IACR,IAAI,IAAI;IACR,IAAI,IAAI;IACR,IAAI,IAAI;IACR,IAAI,KAAK;IACT,MAAM,KAAK;IACX,MAAM,KAAK;IACX,MAAM,KAAK;IACX,MAAM,KAAK;IACX,MAAM,KAAK;IACX,MAAM,KAAK;IACX,MAAM,KAAK;IACX,IAAI,KAAK;IACT,OAAO,KAAK;IACZ,MAAM,KAAK;IACX,MAAM,KAAK;IACX,MAAM,KAAK;IACX,OAAO,KAAK;IACZ,KAAK,KAAK;IACV,MAAM,KAAK;IACX,KAAK,KAAK;IACV,KAAK,KAAK;IACV,KAAK,KAAK;IACV,KAAK,KAAK;IACV,MAAM,KAAK;IACX,KAAK,KAAK;IACV,IAAI,KAAK;IACT,QAAQ,KAAK;IACb,QAAQ,KAAK;IACb,QAAQ,KAAK;IACb,KAAK,KAAK;IACV,KAAK,KAAK;IACV,SAAS,KAAK;IACd,KAAK,KAAK;IACV,IAAI,KAAK;IACT,IAAI,KAAK;IAIT,OAAO,OAAO;IACd,OAAO,OAAO;IACd,OAAO,OAAO;IACd,OAAO,OAAO;IACd,OAAO,OAAO;IACd,OAAO,OAAO;CACd;AAGD,eAAO,MAAM,aAAa,QAEzB,CAAC;AACF,eAAO,MAAM,oBAAoB,QAAiC,CAAC;AAEnE,wBAAgB,mBAAmB,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS,CAGrE;AAKD,eAAO,MAAM,gBAAgB,EAAE,wBAAwB,EA6DtD,CAAC;AAIF,wBAAgB,oBAAoB,CACnC,KAAK,EAAE,wBAAwB,EAC/B,eAAe,EAAE,wBAAwB,EAAE,GACzC,wBAAwB,GAAG,SAAS,CAmCtC;AAGD,oBAAY,oBAAoB;IAC/B,GAAG,IAAI;IACP,GAAG,IAAI;IACP,IAAI,IAAI;IACR,IAAI,IAAI;IACR,IAAI,IAAI;IACR,IAAI,IAAI;IACR,IAAI,IAAI;IACR,IAAI,IAAI;IACR,IAAI,KAAK;IACT,IAAI,KAAK;IACT,IAAI,KAAK;IACT,IAAI,KAAK;IACT,IAAI,KAAK;IACT,IAAI,KAAK;IACT,OAAO,KAAK;IACZ,MAAM,KAAK;IACX,OAAO,KAAK;IACZ,KAAK,KAAK;IACV,MAAM,KAAK;IACX,KAAK,KAAK;IACV,KAAK,KAAK;IACV,MAAM,KAAK;IACX,EAAE,KAAK;IACP,GAAG,KAAK;IACR,GAAG,KAAK;IACR,GAAG,KAAK;IACR,GAAG,KAAK;IACR,KAAK,KAAK;IACV,IAAI,KAAK;IACT,KAAK,KAAK;IACV,KAAK,KAAK;IACV,KAAK,KAAK;IACV,KAAK,KAAK;IACV,IAAI,KAAK;IACT,IAAI,KAAK;CACT"}
|
package/dist/esm/gguf.js
CHANGED
|
@@ -44,6 +44,7 @@ export var GGMLFileQuantizationType;
|
|
|
44
44
|
GGMLFileQuantizationType[GGMLFileQuantizationType["MXFP4_MOE"] = 38] = "MXFP4_MOE";
|
|
45
45
|
GGMLFileQuantizationType[GGMLFileQuantizationType["NVFP4"] = 39] = "NVFP4";
|
|
46
46
|
GGMLFileQuantizationType[GGMLFileQuantizationType["Q1_0"] = 40] = "Q1_0";
|
|
47
|
+
GGMLFileQuantizationType[GGMLFileQuantizationType["Q2_0"] = 41] = "Q2_0";
|
|
47
48
|
// custom quants used by unsloth
|
|
48
49
|
// they are not officially a scheme enum value in GGUF, but only here for naming
|
|
49
50
|
GGMLFileQuantizationType[GGMLFileQuantizationType["Q2_K_XL"] = 1000] = "Q2_K_XL";
|
|
@@ -111,6 +112,7 @@ export const GGUF_QUANT_ORDER = [
|
|
|
111
112
|
GGMLFileQuantizationType.IQ2_S,
|
|
112
113
|
GGMLFileQuantizationType.IQ2_XS,
|
|
113
114
|
GGMLFileQuantizationType.IQ2_XXS,
|
|
115
|
+
GGMLFileQuantizationType.Q2_0,
|
|
114
116
|
// 1-bit quantizations
|
|
115
117
|
GGMLFileQuantizationType.IQ1_S,
|
|
116
118
|
GGMLFileQuantizationType.IQ1_M,
|
|
@@ -188,4 +190,5 @@ export var GGMLQuantizationType;
|
|
|
188
190
|
GGMLQuantizationType[GGMLQuantizationType["MXFP4"] = 39] = "MXFP4";
|
|
189
191
|
GGMLQuantizationType[GGMLQuantizationType["NVFP4"] = 40] = "NVFP4";
|
|
190
192
|
GGMLQuantizationType[GGMLQuantizationType["Q1_0"] = 41] = "Q1_0";
|
|
193
|
+
GGMLQuantizationType[GGMLQuantizationType["Q2_0"] = 42] = "Q2_0";
|
|
191
194
|
})(GGMLQuantizationType || (GGMLQuantizationType = {}));
|
|
@@ -359,7 +359,7 @@ const diffusers_default = (model) => [
|
|
|
359
359
|
from diffusers import DiffusionPipeline
|
|
360
360
|
|
|
361
361
|
# switch to "mps" for apple devices
|
|
362
|
-
pipe = DiffusionPipeline.from_pretrained("${model.id}",
|
|
362
|
+
pipe = DiffusionPipeline.from_pretrained("${model.id}", torch_dtype=torch.bfloat16, device_map="cuda")
|
|
363
363
|
|
|
364
364
|
prompt = "${get_prompt_from_diffusers_model(model) ?? diffusersDefaultPrompt}"
|
|
365
365
|
image = pipe(prompt).images[0]`,
|
|
@@ -370,7 +370,7 @@ from diffusers import DiffusionPipeline
|
|
|
370
370
|
from diffusers.utils import load_image
|
|
371
371
|
|
|
372
372
|
# switch to "mps" for apple devices
|
|
373
|
-
pipe = DiffusionPipeline.from_pretrained("${model.id}",
|
|
373
|
+
pipe = DiffusionPipeline.from_pretrained("${model.id}", torch_dtype=torch.bfloat16, device_map="cuda")
|
|
374
374
|
|
|
375
375
|
prompt = "${get_prompt_from_diffusers_model(model) ?? diffusersImg2ImgDefaultPrompt}"
|
|
376
376
|
input_image = load_image("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/cat.png")
|
|
@@ -383,7 +383,7 @@ from diffusers import DiffusionPipeline
|
|
|
383
383
|
from diffusers.utils import load_image, export_to_video
|
|
384
384
|
|
|
385
385
|
# switch to "mps" for apple devices
|
|
386
|
-
pipe = DiffusionPipeline.from_pretrained("${model.id}",
|
|
386
|
+
pipe = DiffusionPipeline.from_pretrained("${model.id}", torch_dtype=torch.bfloat16, device_map="cuda")
|
|
387
387
|
pipe.to("cuda")
|
|
388
388
|
|
|
389
389
|
prompt = "${get_prompt_from_diffusers_model(model) ?? diffusersVideoDefaultPrompt}"
|
|
@@ -407,7 +407,7 @@ const diffusers_lora = (model) => [
|
|
|
407
407
|
from diffusers import DiffusionPipeline
|
|
408
408
|
|
|
409
409
|
# switch to "mps" for apple devices
|
|
410
|
-
pipe = DiffusionPipeline.from_pretrained("${get_base_diffusers_model(model)}",
|
|
410
|
+
pipe = DiffusionPipeline.from_pretrained("${get_base_diffusers_model(model)}", torch_dtype=torch.bfloat16, device_map="cuda")
|
|
411
411
|
pipe.load_lora_weights("${model.id}")
|
|
412
412
|
|
|
413
413
|
prompt = "${get_prompt_from_diffusers_model(model) ?? diffusersDefaultPrompt}"
|
|
@@ -419,7 +419,7 @@ from diffusers import DiffusionPipeline
|
|
|
419
419
|
from diffusers.utils import load_image
|
|
420
420
|
|
|
421
421
|
# switch to "mps" for apple devices
|
|
422
|
-
pipe = DiffusionPipeline.from_pretrained("${get_base_diffusers_model(model)}",
|
|
422
|
+
pipe = DiffusionPipeline.from_pretrained("${get_base_diffusers_model(model)}", torch_dtype=torch.bfloat16, device_map="cuda")
|
|
423
423
|
pipe.load_lora_weights("${model.id}")
|
|
424
424
|
|
|
425
425
|
prompt = "${get_prompt_from_diffusers_model(model) ?? diffusersImg2ImgDefaultPrompt}"
|
|
@@ -433,7 +433,7 @@ from diffusers import DiffusionPipeline
|
|
|
433
433
|
from diffusers.utils import export_to_video
|
|
434
434
|
|
|
435
435
|
# switch to "mps" for apple devices
|
|
436
|
-
pipe = DiffusionPipeline.from_pretrained("${get_base_diffusers_model(model)}",
|
|
436
|
+
pipe = DiffusionPipeline.from_pretrained("${get_base_diffusers_model(model)}", torch_dtype=torch.bfloat16, device_map="cuda")
|
|
437
437
|
pipe.load_lora_weights("${model.id}")
|
|
438
438
|
|
|
439
439
|
prompt = "${get_prompt_from_diffusers_model(model) ?? diffusersVideoDefaultPrompt}"
|
|
@@ -447,7 +447,7 @@ from diffusers import DiffusionPipeline
|
|
|
447
447
|
from diffusers.utils import load_image, export_to_video
|
|
448
448
|
|
|
449
449
|
# switch to "mps" for apple devices
|
|
450
|
-
pipe = DiffusionPipeline.from_pretrained("${get_base_diffusers_model(model)}",
|
|
450
|
+
pipe = DiffusionPipeline.from_pretrained("${get_base_diffusers_model(model)}", torch_dtype=torch.bfloat16, device_map="cuda")
|
|
451
451
|
pipe.load_lora_weights("${model.id}")
|
|
452
452
|
|
|
453
453
|
prompt = "${get_prompt_from_diffusers_model(model) ?? diffusersVideoDefaultPrompt}"
|
|
@@ -461,7 +461,7 @@ const diffusers_textual_inversion = (model) => [
|
|
|
461
461
|
from diffusers import DiffusionPipeline
|
|
462
462
|
|
|
463
463
|
# switch to "mps" for apple devices
|
|
464
|
-
pipe = DiffusionPipeline.from_pretrained("${get_base_diffusers_model(model)}",
|
|
464
|
+
pipe = DiffusionPipeline.from_pretrained("${get_base_diffusers_model(model)}", torch_dtype=torch.bfloat16, device_map="cuda")
|
|
465
465
|
pipe.load_textual_inversion("${model.id}")`,
|
|
466
466
|
];
|
|
467
467
|
const diffusers_flux_fill = (model) => [
|
|
@@ -473,7 +473,7 @@ image = load_image("https://huggingface.co/datasets/diffusers/diffusers-images-d
|
|
|
473
473
|
mask = load_image("https://huggingface.co/datasets/diffusers/diffusers-images-docs/resolve/main/cup_mask.png")
|
|
474
474
|
|
|
475
475
|
# switch to "mps" for apple devices
|
|
476
|
-
pipe = FluxFillPipeline.from_pretrained("${model.id}",
|
|
476
|
+
pipe = FluxFillPipeline.from_pretrained("${model.id}", torch_dtype=torch.bfloat16, device_map="cuda")
|
|
477
477
|
image = pipe(
|
|
478
478
|
prompt="a white paper cup",
|
|
479
479
|
image=image,
|
|
@@ -493,7 +493,7 @@ from diffusers import AutoPipelineForInpainting
|
|
|
493
493
|
from diffusers.utils import load_image
|
|
494
494
|
|
|
495
495
|
# switch to "mps" for apple devices
|
|
496
|
-
pipe = AutoPipelineForInpainting.from_pretrained("${model.id}",
|
|
496
|
+
pipe = AutoPipelineForInpainting.from_pretrained("${model.id}", torch_dtype=torch.float16, variant="fp16", device_map="cuda")
|
|
497
497
|
|
|
498
498
|
img_url = "https://raw.githubusercontent.com/CompVis/latent-diffusion/main/data/inpainting_examples/overture-creations-5sI6fQgYIuo.png"
|
|
499
499
|
mask_url = "https://raw.githubusercontent.com/CompVis/latent-diffusion/main/data/inpainting_examples/overture-creations-5sI6fQgYIuo_mask.png"
|
|
@@ -1672,7 +1672,7 @@ export const transformers = (model) => {
|
|
|
1672
1672
|
: info.processor === "AutoFeatureExtractor"
|
|
1673
1673
|
? "extractor"
|
|
1674
1674
|
: "processor";
|
|
1675
|
-
autoSnippet.push("# Load model directly", `from transformers import ${info.processor}, ${info.auto_model}`, "", `${processorVarName} = ${info.processor}.from_pretrained("${model.id}"` + remote_code_snippet + ")", `model = ${info.auto_model}.from_pretrained("${model.id}"` + remote_code_snippet + "
|
|
1675
|
+
autoSnippet.push("# Load model directly", `from transformers import ${info.processor}, ${info.auto_model}`, "", `${processorVarName} = ${info.processor}.from_pretrained("${model.id}"` + remote_code_snippet + ")", `model = ${info.auto_model}.from_pretrained("${model.id}"` + remote_code_snippet + ', device_map="auto")');
|
|
1676
1676
|
if (model.tags.includes("conversational") && hasChatTemplate(model)) {
|
|
1677
1677
|
if (model.tags.includes("image-text-to-text")) {
|
|
1678
1678
|
autoSnippet.push("messages = [", [
|
|
@@ -1692,7 +1692,7 @@ export const transformers = (model) => {
|
|
|
1692
1692
|
}
|
|
1693
1693
|
}
|
|
1694
1694
|
else {
|
|
1695
|
-
autoSnippet.push("# Load model directly", `from transformers import ${info.auto_model}`, `model = ${info.auto_model}.from_pretrained("${model.id}"` + remote_code_snippet + ',
|
|
1695
|
+
autoSnippet.push("# Load model directly", `from transformers import ${info.auto_model}`, `model = ${info.auto_model}.from_pretrained("${model.id}"` + remote_code_snippet + ', device_map="auto")');
|
|
1696
1696
|
}
|
|
1697
1697
|
if (model.pipeline_tag && LIBRARY_TASK_MAPPING.transformers?.includes(model.pipeline_tag)) {
|
|
1698
1698
|
const pipelineSnippet = ["# Use a pipeline as a high-level helper"];
|
|
@@ -217,6 +217,13 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
|
|
|
217
217
|
repoUrl: string;
|
|
218
218
|
snippets: (model: ModelData) => string[];
|
|
219
219
|
};
|
|
220
|
+
ccpfn: {
|
|
221
|
+
prettyLabel: string;
|
|
222
|
+
repoName: string;
|
|
223
|
+
repoUrl: string;
|
|
224
|
+
filter: false;
|
|
225
|
+
countDownloads: string;
|
|
226
|
+
};
|
|
220
227
|
champ: {
|
|
221
228
|
prettyLabel: string;
|
|
222
229
|
repoName: string;
|
|
@@ -1739,5 +1746,5 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
|
|
|
1739
1746
|
};
|
|
1740
1747
|
export type ModelLibraryKey = keyof typeof MODEL_LIBRARIES_UI_ELEMENTS;
|
|
1741
1748
|
export declare const ALL_MODEL_LIBRARY_KEYS: ModelLibraryKey[];
|
|
1742
|
-
export declare const ALL_DISPLAY_MODEL_LIBRARY_KEYS: ("acestep" | "adapter-transformers" | "allennlp" | "anemoi" | "araclip" | "aviation-ner" | "asteroid" | "audiocraft" | "audioseal" | "bagel-mot" | "bboxmaskpose" | "ben2" | "bertopic" | "big_vision" | "bionemo" | "birder" | "birefnet" | "bm25s" | "boltzgen" | "cancertathomev2" | "cartesia_pytorch" | "cartesia_mlx" | "champ" | "chatterbox" | "chaossim" | "chat_tts" | "chexmix" | "chronos-forecasting" | "clara" | "clipscope" | "cloud-agents" | "collectorvision" | "colipri" | "cosyvoice" | "cotracker" | "colpali" | "comet" | "cosmos" | "cxr-foundation" | "deepforest" | "depth-anything-v2" | "depth-pro" | "derm-foundation" | "describe-anything" | "dia-tts" | "dia2" | "diff-interpretation-tuning" | "diffree" | "diffusers" | "diffusionkit" | "docking-at-home" | "doctr" | "edsnlp" | "elm" | "encoderfile" | "espnet" | "eupe" | "fairseq" | "fastai" | "fastprint" | "fasttext" | "fixer" | "flair" | "flextab" | "fme" | "gemma.cpp" | "geometry-crafter" | "gliner" | "gliner2" | "glm-tts" | "glyph-byt5" | "granite-library" | "grok" | "habibi-tts" | "hallo" | "hermes" | "holomotion" | "hezar" | "htrflow" | "hunyuan-dit" | "hunyuan3d-2" | "hunyuanworld-voyager" | "hy-worldplay" | "hy-world-2" | "image-matching-models" | "imstoucan" | "index-tts" | "infinitetalk" | "infinite-you" | "intellifold" | "ising-decoding" | "keras" | "tf-keras" | "keras-hub" | "kernels" | "kimi-audio" | "kittentts" | "kronos" | "k2" | "lyra-2.0" | "lagernvs" | "lightning-ir" | "litert" | "litert-lm" | "lerobot" | "lightglue" | "liveportrait" | "longcat-video-avatar-1.5" | "llama-cpp-python" | "mini-omni2" | "mindspore" | "magi-1" | "magenta-realtime" | "magenta-realtime-2" | "mamba-ssm" | "manas-1" | "mars5-tts" | "matanyone" | "mesh-anything" | "merlin" | "medvae" | "mitie" | "ml-agents" | "ml-sharp" | "mlx" | "mlx-image" | "mlc-llm" | "model2vec" | "mobilint" | "moshi" | "mtvcraft" | "multimolecule" | "nemo" | "nv-medtech" | "open-oasis" | "open_clip" | "openasr" | "openpeerllm" | "open-sora" | "outetts" | "paddlenlp" | "PaddleOCR" | "peft" | "perception-encoder" | "phantom-wan" | "pocket-tts" | "pruna-ai" | "pxia" | "pyannote-audio" | "py-feat" | "pythae" | "quantumpeer" | "qwen3_tts" | "recurrentgemma" | "relik" | "refiners" | "renderformer" | "reverb" | "rkllm" | "robo-orchard-lab" | "rwkv" | "saelens" | "scail-2" | "sam2" | "sam-3d-body" | "sam-3d-objects" | "same" | "sample-factory" | "sap-rpt-1-oss" | "sapiens" | "sapiens2" | "seedvr" | "self-forcing" | "sentence-transformers" | "setfit" | "sklearn" | "spacy" | "span-marker" | "speechbrain" | "ssr-speech" | "stable-audio-3" | "stable-audio-tools" | "monkeyocr" | "diffusion-single-file" | "seed-story" | "skala" | "soloaudio" | "songbloom" | "stable-baselines3" | "stanza" | "supertonic" | "swarmformer" | "synthefy-migas" | "f5-tts" | "genmo" | "tencent-song-generation" | "tensorflowtts" | "tensorrt" | "tabpfn" | "terratorch" | "tic-clip" | "timesfm" | "timee-ts" | "timm" | "tirex" | "tirex-2" | "torchgeo" | "transformers" | "transformers.js" | "trellis" | "trellis2" | "tunejury" | "ultralytics" | "univa" | "uni-3dar" | "unity-sentis" | "sana" | "videoprism" | "vfi-mamba" | "vismatch" | "lvface" | "voicecraft" | "voxcpm" | "vui" | "vibevoice" | "videox_fun" | "wan2.2" | "wham" | "whisperkit" | "yolov10" | "yolov26" | "zonos" | "3dtopia-xl")[];
|
|
1749
|
+
export declare const ALL_DISPLAY_MODEL_LIBRARY_KEYS: ("acestep" | "adapter-transformers" | "allennlp" | "anemoi" | "araclip" | "aviation-ner" | "asteroid" | "audiocraft" | "audioseal" | "bagel-mot" | "bboxmaskpose" | "ben2" | "bertopic" | "big_vision" | "bionemo" | "birder" | "birefnet" | "bm25s" | "boltzgen" | "cancertathomev2" | "cartesia_pytorch" | "cartesia_mlx" | "ccpfn" | "champ" | "chatterbox" | "chaossim" | "chat_tts" | "chexmix" | "chronos-forecasting" | "clara" | "clipscope" | "cloud-agents" | "collectorvision" | "colipri" | "cosyvoice" | "cotracker" | "colpali" | "comet" | "cosmos" | "cxr-foundation" | "deepforest" | "depth-anything-v2" | "depth-pro" | "derm-foundation" | "describe-anything" | "dia-tts" | "dia2" | "diff-interpretation-tuning" | "diffree" | "diffusers" | "diffusionkit" | "docking-at-home" | "doctr" | "edsnlp" | "elm" | "encoderfile" | "espnet" | "eupe" | "fairseq" | "fastai" | "fastprint" | "fasttext" | "fixer" | "flair" | "flextab" | "fme" | "gemma.cpp" | "geometry-crafter" | "gliner" | "gliner2" | "glm-tts" | "glyph-byt5" | "granite-library" | "grok" | "habibi-tts" | "hallo" | "hermes" | "holomotion" | "hezar" | "htrflow" | "hunyuan-dit" | "hunyuan3d-2" | "hunyuanworld-voyager" | "hy-worldplay" | "hy-world-2" | "image-matching-models" | "imstoucan" | "index-tts" | "infinitetalk" | "infinite-you" | "intellifold" | "ising-decoding" | "keras" | "tf-keras" | "keras-hub" | "kernels" | "kimi-audio" | "kittentts" | "kronos" | "k2" | "lyra-2.0" | "lagernvs" | "lightning-ir" | "litert" | "litert-lm" | "lerobot" | "lightglue" | "liveportrait" | "longcat-video-avatar-1.5" | "llama-cpp-python" | "mini-omni2" | "mindspore" | "magi-1" | "magenta-realtime" | "magenta-realtime-2" | "mamba-ssm" | "manas-1" | "mars5-tts" | "matanyone" | "mesh-anything" | "merlin" | "medvae" | "mitie" | "ml-agents" | "ml-sharp" | "mlx" | "mlx-image" | "mlc-llm" | "model2vec" | "mobilint" | "moshi" | "mtvcraft" | "multimolecule" | "nemo" | "nv-medtech" | "open-oasis" | "open_clip" | "openasr" | "openpeerllm" | "open-sora" | "outetts" | "paddlenlp" | "PaddleOCR" | "peft" | "perception-encoder" | "phantom-wan" | "pocket-tts" | "pruna-ai" | "pxia" | "pyannote-audio" | "py-feat" | "pythae" | "quantumpeer" | "qwen3_tts" | "recurrentgemma" | "relik" | "refiners" | "renderformer" | "reverb" | "rkllm" | "robo-orchard-lab" | "rwkv" | "saelens" | "scail-2" | "sam2" | "sam-3d-body" | "sam-3d-objects" | "same" | "sample-factory" | "sap-rpt-1-oss" | "sapiens" | "sapiens2" | "seedvr" | "self-forcing" | "sentence-transformers" | "setfit" | "sklearn" | "spacy" | "span-marker" | "speechbrain" | "ssr-speech" | "stable-audio-3" | "stable-audio-tools" | "monkeyocr" | "diffusion-single-file" | "seed-story" | "skala" | "soloaudio" | "songbloom" | "stable-baselines3" | "stanza" | "supertonic" | "swarmformer" | "synthefy-migas" | "f5-tts" | "genmo" | "tencent-song-generation" | "tensorflowtts" | "tensorrt" | "tabpfn" | "terratorch" | "tic-clip" | "timesfm" | "timee-ts" | "timm" | "tirex" | "tirex-2" | "torchgeo" | "transformers" | "transformers.js" | "trellis" | "trellis2" | "tunejury" | "ultralytics" | "univa" | "uni-3dar" | "unity-sentis" | "sana" | "videoprism" | "vfi-mamba" | "vismatch" | "lvface" | "voicecraft" | "voxcpm" | "vui" | "vibevoice" | "videox_fun" | "wan2.2" | "wham" | "whisperkit" | "yolov10" | "yolov26" | "zonos" | "3dtopia-xl")[];
|
|
1743
1750
|
//# sourceMappingURL=model-libraries.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"model-libraries.d.ts","sourceRoot":"","sources":["../../src/model-libraries.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AACjD,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,gCAAgC,CAAC;AAEzE;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAChC;;;;OAIG;IACH,WAAW,EAAE,MAAM,CAAC;IACpB;;OAEG;IACH,QAAQ,EAAE,MAAM,CAAC;IACjB;;OAEG;IACH,OAAO,EAAE,MAAM,CAAC;IAChB;;OAEG;IACH,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB;;OAEG;IACH,QAAQ,CAAC,EAAE,CAAC,KAAK,EAAE,SAAS,KAAK,MAAM,EAAE,CAAC;IAC1C;;;;;OAKG;IACH,cAAc,CAAC,EAAE,kBAAkB,CAAC;IACpC;;;OAGG;IACH,MAAM,CAAC,EAAE,OAAO,CAAC;CACjB;AAED;;;;;;;;;;;;;GAaG;AAEH,eAAO,MAAM,2BAA2B
|
|
1
|
+
{"version":3,"file":"model-libraries.d.ts","sourceRoot":"","sources":["../../src/model-libraries.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AACjD,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,gCAAgC,CAAC;AAEzE;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAChC;;;;OAIG;IACH,WAAW,EAAE,MAAM,CAAC;IACpB;;OAEG;IACH,QAAQ,EAAE,MAAM,CAAC;IACjB;;OAEG;IACH,OAAO,EAAE,MAAM,CAAC;IAChB;;OAEG;IACH,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB;;OAEG;IACH,QAAQ,CAAC,EAAE,CAAC,KAAK,EAAE,SAAS,KAAK,MAAM,EAAE,CAAC;IAC1C;;;;;OAKG;IACH,cAAc,CAAC,EAAE,kBAAkB,CAAC;IACpC;;;OAGG;IACH,MAAM,CAAC,EAAE,OAAO,CAAC;CACjB;AAED;;;;;;;;;;;;;GAaG;AAEH,eAAO,MAAM,2BAA2B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAiqDI,CAAC;AAE7C,MAAM,MAAM,eAAe,GAAG,MAAM,OAAO,2BAA2B,CAAC;AAEvE,eAAO,MAAM,sBAAsB,EAA+C,eAAe,EAAE,CAAC;AAEpG,eAAO,MAAM,8BAA8B,+vGAQ1B,CAAC"}
|
|
@@ -177,6 +177,13 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
|
|
|
177
177
|
repoUrl: "https://github.com/cartesia-ai/cartesia_mlx",
|
|
178
178
|
snippets: snippets.cartesia_mlx,
|
|
179
179
|
},
|
|
180
|
+
ccpfn: {
|
|
181
|
+
prettyLabel: "CCPFN",
|
|
182
|
+
repoName: "CCPFN",
|
|
183
|
+
repoUrl: "https://huggingface.co/Layer6/CCPFN",
|
|
184
|
+
filter: false,
|
|
185
|
+
countDownloads: `path_extension:"pt"`,
|
|
186
|
+
},
|
|
180
187
|
champ: {
|
|
181
188
|
prettyLabel: "Champ",
|
|
182
189
|
repoName: "Champ",
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"inputs.d.ts","sourceRoot":"","sources":["../../../src/snippets/inputs.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,0BAA0B,EAAE,MAAM,mBAAmB,CAAC;AACpE,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;
|
|
1
|
+
{"version":3,"file":"inputs.d.ts","sourceRoot":"","sources":["../../../src/snippets/inputs.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,0BAA0B,EAAE,MAAM,mBAAmB,CAAC;AACpE,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAyKnD,wBAAgB,oBAAoB,CACnC,KAAK,EAAE,gBAAgB,EACvB,MAAM,UAAQ,EACd,QAAQ,UAAQ,GACd,MAAM,GAAG,0BAA0B,EAAE,CAmBvC"}
|
|
@@ -48,7 +48,10 @@ const inputsTextGeneration = (model) => {
|
|
|
48
48
|
}
|
|
49
49
|
return `"Can you please let us know more details about your "`;
|
|
50
50
|
};
|
|
51
|
-
|
|
51
|
+
// `mask_token` is author-controlled (tokenizer/config.json); build the string then
|
|
52
|
+
// JSON.stringify it so quotes/backslashes are escaped into a valid string literal
|
|
53
|
+
// and cannot break out of the generated snippet.
|
|
54
|
+
const inputsFillMask = (model) => JSON.stringify(`The answer to the universe is ${model.mask_token}.`);
|
|
52
55
|
const inputsSentenceSimilarity = () => `{
|
|
53
56
|
"source_sentence": "That is a happy person",
|
|
54
57
|
"sentences": [
|
package/package.json
CHANGED
package/src/gguf.ts
CHANGED
|
@@ -43,6 +43,7 @@ export enum GGMLFileQuantizationType {
|
|
|
43
43
|
MXFP4_MOE = 38,
|
|
44
44
|
NVFP4 = 39,
|
|
45
45
|
Q1_0 = 40,
|
|
46
|
+
Q2_0 = 41,
|
|
46
47
|
|
|
47
48
|
// custom quants used by unsloth
|
|
48
49
|
// they are not officially a scheme enum value in GGUF, but only here for naming
|
|
@@ -121,6 +122,7 @@ export const GGUF_QUANT_ORDER: GGMLFileQuantizationType[] = [
|
|
|
121
122
|
GGMLFileQuantizationType.IQ2_S,
|
|
122
123
|
GGMLFileQuantizationType.IQ2_XS,
|
|
123
124
|
GGMLFileQuantizationType.IQ2_XXS,
|
|
125
|
+
GGMLFileQuantizationType.Q2_0,
|
|
124
126
|
|
|
125
127
|
// 1-bit quantizations
|
|
126
128
|
GGMLFileQuantizationType.IQ1_S,
|
|
@@ -208,4 +210,5 @@ export enum GGMLQuantizationType {
|
|
|
208
210
|
MXFP4 = 39,
|
|
209
211
|
NVFP4 = 40,
|
|
210
212
|
Q1_0 = 41,
|
|
213
|
+
Q2_0 = 42,
|
|
211
214
|
}
|
|
@@ -405,7 +405,7 @@ const diffusers_default = (model: ModelData) => [
|
|
|
405
405
|
from diffusers import DiffusionPipeline
|
|
406
406
|
|
|
407
407
|
# switch to "mps" for apple devices
|
|
408
|
-
pipe = DiffusionPipeline.from_pretrained("${model.id}",
|
|
408
|
+
pipe = DiffusionPipeline.from_pretrained("${model.id}", torch_dtype=torch.bfloat16, device_map="cuda")
|
|
409
409
|
|
|
410
410
|
prompt = "${get_prompt_from_diffusers_model(model) ?? diffusersDefaultPrompt}"
|
|
411
411
|
image = pipe(prompt).images[0]`,
|
|
@@ -417,7 +417,7 @@ from diffusers import DiffusionPipeline
|
|
|
417
417
|
from diffusers.utils import load_image
|
|
418
418
|
|
|
419
419
|
# switch to "mps" for apple devices
|
|
420
|
-
pipe = DiffusionPipeline.from_pretrained("${model.id}",
|
|
420
|
+
pipe = DiffusionPipeline.from_pretrained("${model.id}", torch_dtype=torch.bfloat16, device_map="cuda")
|
|
421
421
|
|
|
422
422
|
prompt = "${get_prompt_from_diffusers_model(model) ?? diffusersImg2ImgDefaultPrompt}"
|
|
423
423
|
input_image = load_image("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/cat.png")
|
|
@@ -431,7 +431,7 @@ from diffusers import DiffusionPipeline
|
|
|
431
431
|
from diffusers.utils import load_image, export_to_video
|
|
432
432
|
|
|
433
433
|
# switch to "mps" for apple devices
|
|
434
|
-
pipe = DiffusionPipeline.from_pretrained("${model.id}",
|
|
434
|
+
pipe = DiffusionPipeline.from_pretrained("${model.id}", torch_dtype=torch.bfloat16, device_map="cuda")
|
|
435
435
|
pipe.to("cuda")
|
|
436
436
|
|
|
437
437
|
prompt = "${get_prompt_from_diffusers_model(model) ?? diffusersVideoDefaultPrompt}"
|
|
@@ -457,7 +457,7 @@ const diffusers_lora = (model: ModelData) => [
|
|
|
457
457
|
from diffusers import DiffusionPipeline
|
|
458
458
|
|
|
459
459
|
# switch to "mps" for apple devices
|
|
460
|
-
pipe = DiffusionPipeline.from_pretrained("${get_base_diffusers_model(model)}",
|
|
460
|
+
pipe = DiffusionPipeline.from_pretrained("${get_base_diffusers_model(model)}", torch_dtype=torch.bfloat16, device_map="cuda")
|
|
461
461
|
pipe.load_lora_weights("${model.id}")
|
|
462
462
|
|
|
463
463
|
prompt = "${get_prompt_from_diffusers_model(model) ?? diffusersDefaultPrompt}"
|
|
@@ -470,7 +470,7 @@ from diffusers import DiffusionPipeline
|
|
|
470
470
|
from diffusers.utils import load_image
|
|
471
471
|
|
|
472
472
|
# switch to "mps" for apple devices
|
|
473
|
-
pipe = DiffusionPipeline.from_pretrained("${get_base_diffusers_model(model)}",
|
|
473
|
+
pipe = DiffusionPipeline.from_pretrained("${get_base_diffusers_model(model)}", torch_dtype=torch.bfloat16, device_map="cuda")
|
|
474
474
|
pipe.load_lora_weights("${model.id}")
|
|
475
475
|
|
|
476
476
|
prompt = "${get_prompt_from_diffusers_model(model) ?? diffusersImg2ImgDefaultPrompt}"
|
|
@@ -485,7 +485,7 @@ from diffusers import DiffusionPipeline
|
|
|
485
485
|
from diffusers.utils import export_to_video
|
|
486
486
|
|
|
487
487
|
# switch to "mps" for apple devices
|
|
488
|
-
pipe = DiffusionPipeline.from_pretrained("${get_base_diffusers_model(model)}",
|
|
488
|
+
pipe = DiffusionPipeline.from_pretrained("${get_base_diffusers_model(model)}", torch_dtype=torch.bfloat16, device_map="cuda")
|
|
489
489
|
pipe.load_lora_weights("${model.id}")
|
|
490
490
|
|
|
491
491
|
prompt = "${get_prompt_from_diffusers_model(model) ?? diffusersVideoDefaultPrompt}"
|
|
@@ -500,7 +500,7 @@ from diffusers import DiffusionPipeline
|
|
|
500
500
|
from diffusers.utils import load_image, export_to_video
|
|
501
501
|
|
|
502
502
|
# switch to "mps" for apple devices
|
|
503
|
-
pipe = DiffusionPipeline.from_pretrained("${get_base_diffusers_model(model)}",
|
|
503
|
+
pipe = DiffusionPipeline.from_pretrained("${get_base_diffusers_model(model)}", torch_dtype=torch.bfloat16, device_map="cuda")
|
|
504
504
|
pipe.load_lora_weights("${model.id}")
|
|
505
505
|
|
|
506
506
|
prompt = "${get_prompt_from_diffusers_model(model) ?? diffusersVideoDefaultPrompt}"
|
|
@@ -515,7 +515,7 @@ const diffusers_textual_inversion = (model: ModelData) => [
|
|
|
515
515
|
from diffusers import DiffusionPipeline
|
|
516
516
|
|
|
517
517
|
# switch to "mps" for apple devices
|
|
518
|
-
pipe = DiffusionPipeline.from_pretrained("${get_base_diffusers_model(model)}",
|
|
518
|
+
pipe = DiffusionPipeline.from_pretrained("${get_base_diffusers_model(model)}", torch_dtype=torch.bfloat16, device_map="cuda")
|
|
519
519
|
pipe.load_textual_inversion("${model.id}")`,
|
|
520
520
|
];
|
|
521
521
|
|
|
@@ -528,7 +528,7 @@ image = load_image("https://huggingface.co/datasets/diffusers/diffusers-images-d
|
|
|
528
528
|
mask = load_image("https://huggingface.co/datasets/diffusers/diffusers-images-docs/resolve/main/cup_mask.png")
|
|
529
529
|
|
|
530
530
|
# switch to "mps" for apple devices
|
|
531
|
-
pipe = FluxFillPipeline.from_pretrained("${model.id}",
|
|
531
|
+
pipe = FluxFillPipeline.from_pretrained("${model.id}", torch_dtype=torch.bfloat16, device_map="cuda")
|
|
532
532
|
image = pipe(
|
|
533
533
|
prompt="a white paper cup",
|
|
534
534
|
image=image,
|
|
@@ -549,7 +549,7 @@ from diffusers import AutoPipelineForInpainting
|
|
|
549
549
|
from diffusers.utils import load_image
|
|
550
550
|
|
|
551
551
|
# switch to "mps" for apple devices
|
|
552
|
-
pipe = AutoPipelineForInpainting.from_pretrained("${model.id}",
|
|
552
|
+
pipe = AutoPipelineForInpainting.from_pretrained("${model.id}", torch_dtype=torch.float16, variant="fp16", device_map="cuda")
|
|
553
553
|
|
|
554
554
|
img_url = "https://raw.githubusercontent.com/CompVis/latent-diffusion/main/data/inpainting_examples/overture-creations-5sI6fQgYIuo.png"
|
|
555
555
|
mask_url = "https://raw.githubusercontent.com/CompVis/latent-diffusion/main/data/inpainting_examples/overture-creations-5sI6fQgYIuo_mask.png"
|
|
@@ -1835,7 +1835,7 @@ export const transformers = (model: ModelData): string[] => {
|
|
|
1835
1835
|
`from transformers import ${info.processor}, ${info.auto_model}`,
|
|
1836
1836
|
"",
|
|
1837
1837
|
`${processorVarName} = ${info.processor}.from_pretrained("${model.id}"` + remote_code_snippet + ")",
|
|
1838
|
-
`model = ${info.auto_model}.from_pretrained("${model.id}"` + remote_code_snippet + ")
|
|
1838
|
+
`model = ${info.auto_model}.from_pretrained("${model.id}"` + remote_code_snippet + ', device_map="auto")',
|
|
1839
1839
|
);
|
|
1840
1840
|
if (model.tags.includes("conversational") && hasChatTemplate(model)) {
|
|
1841
1841
|
if (model.tags.includes("image-text-to-text")) {
|
|
@@ -1872,7 +1872,7 @@ export const transformers = (model: ModelData): string[] => {
|
|
|
1872
1872
|
autoSnippet.push(
|
|
1873
1873
|
"# Load model directly",
|
|
1874
1874
|
`from transformers import ${info.auto_model}`,
|
|
1875
|
-
`model = ${info.auto_model}.from_pretrained("${model.id}"` + remote_code_snippet + ',
|
|
1875
|
+
`model = ${info.auto_model}.from_pretrained("${model.id}"` + remote_code_snippet + ', device_map="auto")',
|
|
1876
1876
|
);
|
|
1877
1877
|
}
|
|
1878
1878
|
|
package/src/model-libraries.ts
CHANGED
|
@@ -221,6 +221,13 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
|
|
|
221
221
|
repoUrl: "https://github.com/cartesia-ai/cartesia_mlx",
|
|
222
222
|
snippets: snippets.cartesia_mlx,
|
|
223
223
|
},
|
|
224
|
+
ccpfn: {
|
|
225
|
+
prettyLabel: "CCPFN",
|
|
226
|
+
repoName: "CCPFN",
|
|
227
|
+
repoUrl: "https://huggingface.co/Layer6/CCPFN",
|
|
228
|
+
filter: false,
|
|
229
|
+
countDownloads: `path_extension:"pt"`,
|
|
230
|
+
},
|
|
224
231
|
champ: {
|
|
225
232
|
prettyLabel: "Champ",
|
|
226
233
|
repoName: "Champ",
|
package/src/snippets/inputs.ts
CHANGED
|
@@ -66,7 +66,11 @@ const inputsTextGeneration = (model: ModelDataMinimal): string | ChatCompletionI
|
|
|
66
66
|
return `"Can you please let us know more details about your "`;
|
|
67
67
|
};
|
|
68
68
|
|
|
69
|
-
|
|
69
|
+
// `mask_token` is author-controlled (tokenizer/config.json); build the string then
|
|
70
|
+
// JSON.stringify it so quotes/backslashes are escaped into a valid string literal
|
|
71
|
+
// and cannot break out of the generated snippet.
|
|
72
|
+
const inputsFillMask = (model: ModelDataMinimal) =>
|
|
73
|
+
JSON.stringify(`The answer to the universe is ${model.mask_token}.`);
|
|
70
74
|
|
|
71
75
|
const inputsSentenceSimilarity = () =>
|
|
72
76
|
`{
|