diff --git a/.gitignore b/.gitignore index c938cddaf..01b907de2 100644 --- a/.gitignore +++ b/.gitignore @@ -14,7 +14,6 @@ dist/ downloads/ eggs/ .eggs/ -lib/ lib64/ parts/ sdist/ diff --git a/js/src/lib/components/Icons/IconImageToImage.svelte b/js/src/lib/components/Icons/IconImageToImage.svelte new file mode 100644 index 000000000..1ea332d30 --- /dev/null +++ b/js/src/lib/components/Icons/IconImageToImage.svelte @@ -0,0 +1,25 @@ + + + diff --git a/js/src/lib/components/Icons/IconUnconditionalImageGeneration.svelte b/js/src/lib/components/Icons/IconUnconditionalImageGeneration.svelte new file mode 100644 index 000000000..87aec18e6 --- /dev/null +++ b/js/src/lib/components/Icons/IconUnconditionalImageGeneration.svelte @@ -0,0 +1,25 @@ + + + diff --git a/js/src/lib/components/ModelPipelineIcon/ModelPipelineIcon.svelte b/js/src/lib/components/ModelPipelineIcon/ModelPipelineIcon.svelte index 2a14b72a9..1595168b7 100644 --- a/js/src/lib/components/ModelPipelineIcon/ModelPipelineIcon.svelte +++ b/js/src/lib/components/ModelPipelineIcon/ModelPipelineIcon.svelte @@ -27,6 +27,8 @@ import IconVoiceActivityDetection from "../Icons/IconVoiceActivityDetection.svelte"; import IconZeroShotClassification from "../Icons/IconZeroShotClassification.svelte"; import IconReinforcementLearning from "../Icons/IconReinforcementLearning.svelte"; + import IconImageToImage from "../Icons/IconImageToImage.svelte"; + import IconUnconditionalImageGeneration from "../Icons/IconUnconditionalImageGeneration.svelte"; export let classNames = ""; export let pipeline = ""; @@ -61,6 +63,8 @@ "structured-data-classification": IconStructuredDataClassification, "audio-classification": IconAudioClassification, "reinforcement-learning": IconReinforcementLearning, + "image-to-image": IconImageToImage, + "unconditional-image-generation": IconUnconditionalImageGeneration, }; diff --git a/js/src/lib/interfaces/Types.ts b/js/src/lib/interfaces/Types.ts index 0ab71d7b7..c64086f86 100644 --- a/js/src/lib/interfaces/Types.ts +++ b/js/src/lib/interfaces/Types.ts @@ -36,6 +36,8 @@ export enum PipelineType { "image-segmentation" = "Image Segmentation", "text-to-image" = "Text-to-Image", "image-to-text" = "Image-to-Text", + "image-to-image" = "Image-to-Image", + "unconditional-image-generation" = "Unconditional Image Generation", /// others "structured-data-classification" = "Structured Data Classification", "reinforcement-learning" = "Reinforcement Learning", @@ -100,6 +102,8 @@ export const PIPELINE_TAG_MODALITIES: Record = [ "image-segmentation", "text-to-image", "image-to-text", + "image-to-image", + "unconditional-image-generation", /// others "structured-data-classification", "reinforcement-learning", diff --git a/tasks/src/const.ts b/tasks/src/const.ts index d00691d20..c4a052e05 100644 --- a/tasks/src/const.ts +++ b/tasks/src/const.ts @@ -31,6 +31,8 @@ export const TASKS_MODEL_LIBRARIES: Record