Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
File size: 1,360 Bytes
b2ecf7d 6c7ce80 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 |
/**
* Add your new library here.
*
* This is for modeling (= architectures) libraries, not for file formats (like ONNX, etc).
* File formats live in an enum inside the internal codebase.
*/
export enum ModelLibrary {
"adapter-transformers" = "Adapter Transformers",
"allennlp" = "allenNLP",
"asteroid" = "Asteroid",
"bertopic" = "BERTopic",
"diffusers" = "Diffusers",
"doctr" = "docTR",
"espnet" = "ESPnet",
"fairseq" = "Fairseq",
"flair" = "Flair",
"keras" = "Keras",
"k2" = "K2",
"nemo" = "NeMo",
"open_clip" = "OpenCLIP",
"paddlenlp" = "PaddleNLP",
"peft" = "PEFT",
"pyannote-audio" = "pyannote.audio",
"sample-factory" = "Sample Factory",
"sentence-transformers" = "Sentence Transformers",
"sklearn" = "Scikit-learn",
"spacy" = "spaCy",
"span-marker" = "SpanMarker",
"speechbrain" = "speechbrain",
"tensorflowtts" = "TensorFlowTTS",
"timm" = "Timm",
"fastai" = "fastai",
"transformers" = "Transformers",
"transformers.js" = "Transformers.js",
"stanza" = "Stanza",
"fasttext" = "fastText",
"stable-baselines3" = "Stable-Baselines3",
"ml-agents" = "ML-Agents",
"pythae" = "Pythae",
"mindspore" = "MindSpore",
}
export type ModelLibraryKey = keyof typeof ModelLibrary;
export const ALL_DISPLAY_MODEL_LIBRARY_KEYS = Object.keys(ModelLibrary).filter(
(k) => !["doctr", "k2", "mindspore", "tensorflowtts"].includes(k)
);
|