Skip to content

Commit f449c0f

Browse files
committed
add task mapping
1 parent 7657995 commit f449c0f

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

packages/inference/src/lib/getProviderHelper.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -132,6 +132,7 @@ export const PROVIDERS: Record<InferenceProvider, Partial<Record<InferenceTask,
132132
"text-generation": new OvhCloud.OvhCloudTextGenerationTask(),
133133
},
134134
replicate: {
135+
"image-to-image": new Replicate.ReplicateImageToImageTask(),
135136
"text-to-image": new Replicate.ReplicateTextToImageTask(),
136137
"text-to-speech": new Replicate.ReplicateTextToSpeechTask(),
137138
"text-to-video": new Replicate.ReplicateTextToVideoTask(),

0 commit comments

Comments
 (0)