|
|
|
DS_SIZE_CATEGORIES = ["n<1K", "1K<n<10K", "10K<n<100K", "100K<n<1M", "1M<n<10M", "10M<n<100M",
|
|
"100M<n<1B", "1B<n<10B", "10B<n<100B", "100B<n<1T", "n>1T"]
|
|
|
|
SPACE_HARDWARES = ["cpu-basic", "zero-a10g", "cpu-upgrade", "t4-small", "l4x1", "a10g-large", "l40sx1", "a10g-small", "t4-medium", "cpu-xl", "a100-large"]
|
|
|
|
SPACE_STAGES = ["RUNNING", "SLEEPING", "RUNTIME_ERROR", "PAUSED", "BUILD_ERROR", "CONFIG_ERROR", "BUILDING", "APP_STARTING", "RUNNING_APP_STARTING"]
|
|
|
|
PIPELINES = ["text-generation", "text-to-image", "image-text-to-text", "fill-mask", "text-classification", "sentence-similarity", "automatic-speech-recognition",
|
|
"feature-extraction", "text-to-speech", "text2text-generation", "image-to-image", "text-to-video", "zero-shot-image-classification",
|
|
"image-classification", "image-to-video", "image-to-text", "token-classification", "translation", "time-series-forecasting", "audio-classification",
|
|
"object-detection", "audio-text-to-text", "zero-shot-classification", "text-to-audio", "image-segmentation", "image-feature-extraction",
|
|
"video-text-to-text", "image-to-3d", "any-to-any", "question-answering", "text-to-3d", "zero-shot-object-detection", "summarization",
|
|
"document-question-answering", "visual-question-answering", "depth-estimation", "voice-activity-detection", "audio-to-audio", "video-classification"]
|
|
|
|
EMOJIS = {
|
|
"Pipeline": {"text-to-image": "π¨", "image-to-image": "πΌοΈ", "text-to-speech": "π", "automatic-speech-recognition": "π€",
|
|
"text-to-audio": "π΅", "audio-text-to-text": "ποΈ",
|
|
"image-to-video": "ποΈ", "text-to-video": "π¬",
|
|
"text-generation": "π€", "text2text-generation": "π€"},
|
|
"SDK": {"docker": "π³"}
|
|
}
|
|
|