match HF task names

See https://huggingface.co/tasks
huggingface · Jan 14, 2025 · c756f29 · c756f29
1 parent e4b6a37
commit c756f29
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/packages/inference/src/providers/replicate.ts b/packages/inference/src/providers/replicate.ts
@@ -39,15 +39,17 @@ export const REPLICATE_MODEL_IDS: Record<ModelId, ReplicateId> = {
 	"stabilityai/stable-diffusion-xl-base-1.0": "stability-ai/sdxl",
 	"XLabs-AI/flux-controlnet-collections": "xlabs-ai/flux-dev-controlnet",
 
-	/** image tools */
+	/** image-to-image */
 	"Carve/tracer_b7": "lucataco/remove-bg",
+
+	/** image-classification */
 	"Falconsai/nsfw_image_detection": "falcons-ai/nsfw_image_detection",
 
-	/** language */
+	/** text-generation */
 	"ibm-granite/granite-3.0-2b-instruct": "ibm-granite/granite-3.0-2b-instruct",
 	"ibm-granite/granite-3.0-8b-instruct": "ibm-granite/granite-3.0-8b-instruct",
 
-	/** vision */
+	/** image-text-to-text */
 	"allenai/Molmo-7B-D-0924": "zsxkib/molmo-7b",
 	"THUDM/cogvlm2-video-llama3-chat": "chenxwh/cogvlm2-video",