Skip to content

Commit 5efd3dc

Browse files
arabot777arabot777
committed
change wavespeed-ai to wavespeed
1 parent 8e35b64 commit 5efd3dc

File tree

6 files changed

+20
-20
lines changed

6 files changed

+20
-20
lines changed

‎packages/inference/README.md‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -106,7 +106,7 @@ Only a subset of models are supported when requesting third-party providers. You
106106
- [Cerebras supported models](https://huggingface.co/api/partners/cerebras/models)
107107
- [Groq supported models](https://console.groq.com/docs/models)
108108
- [Novita AI supported models](https://huggingface.co/api/partners/novita/models)
109-
- [Wavespeed.ai supported models](https://huggingface.co/api/partners/wavespeed-ai/models)
109+
- [Wavespeed.ai supported models](https://huggingface.co/api/partners/wavespeed/models)
110110
- [Z.ai supported models](https://huggingface.co/api/partners/zai-org/models)
111111

112112
❗**Important note:** To be compatible, the third-party API must adhere to the "standard" shape API we expect on HF model pages for each pipeline task type.

‎packages/inference/src/lib/getProviderHelper.ts‎

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@ import * as Replicate from "../providers/replicate.js";
5252
import * as Sambanova from "../providers/sambanova.js";
5353
import * as Scaleway from "../providers/scaleway.js";
5454
import * as Together from "../providers/together.js";
55-
import * as WavespeedAI from "../providers/wavespeed-ai.js";
55+
import * as Wavespeed from "../providers/wavespeed.js";
5656
import * as Zai from "../providers/zai-org.js";
5757
import type { InferenceProvider, InferenceProviderOrPolicy, InferenceTask } from "../types.js";
5858
import { InferenceClientInputError } from "../errors.js";
@@ -174,10 +174,10 @@ export const PROVIDERS: Record<InferenceProvider, Partial<Record<InferenceTask,
174174
conversational: new Together.TogetherConversationalTask(),
175175
"text-generation": new Together.TogetherTextGenerationTask(),
176176
},
177-
"wavespeed-ai": {
178-
"text-to-image": new WavespeedAI.WavespeedAITextToImageTask(),
179-
"text-to-video": new WavespeedAI.WavespeedAITextToVideoTask(),
180-
"image-to-image": new WavespeedAI.WavespeedAIImageToImageTask(),
177+
wavespeed: {
178+
"text-to-image": new Wavespeed.WavespeedAITextToImageTask(),
179+
"text-to-video": new Wavespeed.WavespeedAITextToVideoTask(),
180+
"image-to-image": new Wavespeed.WavespeedAIImageToImageTask(),
181181
},
182182
"zai-org": {
183183
conversational: new Zai.ZaiConversationalTask(),

‎packages/inference/src/providers/consts.ts‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,6 @@ export const HARDCODED_MODEL_INFERENCE_MAPPING: Record<
3939
sambanova: {},
4040
scaleway: {},
4141
together: {},
42-
"wavespeed-ai": {},
42+
wavespeed: {},
4343
"zai-org": {},
4444
};

‎packages/inference/src/providers/wavespeed-ai.ts‎ renamed to ‎packages/inference/src/providers/wavespeed.ts‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,7 @@ interface WaveSpeedAISubmitTaskResponse {
6565

6666
abstract class WavespeedAITask extends TaskProviderHelper {
6767
constructor(url?: string) {
68-
super("wavespeed-ai", url || WAVESPEEDAI_API_BASE_URL);
68+
super("wavespeed", url || WAVESPEEDAI_API_BASE_URL);
6969
}
7070

7171
makeRoute(params: UrlParams): string {

‎packages/inference/src/types.ts‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -66,7 +66,7 @@ export const INFERENCE_PROVIDERS = [
6666
"sambanova",
6767
"scaleway",
6868
"together",
69-
"wavespeed-ai",
69+
"wavespeed",
7070
"zai-org",
7171
] as const;
7272

‎packages/inference/test/InferenceClient.spec.ts‎

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -2296,30 +2296,30 @@ describe.skip("InferenceClient", () => {
22962296
() => {
22972297
const client = new InferenceClient(env.HF_WAVESPEED_KEY ?? "dummy");
22982298

2299-
HARDCODED_MODEL_INFERENCE_MAPPING["wavespeed-ai"] = {
2299+
HARDCODED_MODEL_INFERENCE_MAPPING["wavespeed"] = {
23002300
"black-forest-labs/FLUX.1-schnell": {
2301-
provider: "wavespeed-ai",
2301+
provider: "wavespeed",
23022302
hfModelId: "black-forest-labs/FLUX.1-schnell",
23032303
providerId: "wavespeed-ai/flux-schnell",
23042304
status: "live",
23052305
task: "text-to-image",
23062306
},
23072307
"Wan-AI/Wan2.1-T2V-14B": {
2308-
provider: "wavespeed-ai",
2308+
provider: "wavespeed",
23092309
hfModelId: "wavespeed-ai/wan-2.1/t2v-480p",
23102310
providerId: "wavespeed-ai/wan-2.1/t2v-480p",
23112311
status: "live",
23122312
task: "text-to-video",
23132313
},
23142314
"HiDream-ai/HiDream-E1-Full": {
2315-
provider: "wavespeed-ai",
2315+
provider: "wavespeed",
23162316
hfModelId: "wavespeed-ai/hidream-e1-full",
23172317
providerId: "wavespeed-ai/hidream-e1-full",
23182318
status: "live",
23192319
task: "image-to-image",
23202320
},
23212321
"openfree/flux-chatgpt-ghibli-lora": {
2322-
provider: "wavespeed-ai",
2322+
provider: "wavespeed",
23232323
hfModelId: "openfree/flux-chatgpt-ghibli-lora",
23242324
providerId: "wavespeed-ai/flux-dev-lora",
23252325
status: "live",
@@ -2328,7 +2328,7 @@ describe.skip("InferenceClient", () => {
23282328
adapterWeightsPath: "flux-chatgpt-ghibli-lora.safetensors",
23292329
},
23302330
"linoyts/yarn_art_Flux_LoRA": {
2331-
provider: "wavespeed-ai",
2331+
provider: "wavespeed",
23322332
hfModelId: "linoyts/yarn_art_Flux_LoRA",
23332333
providerId: "wavespeed-ai/flux-dev-lora-ultra-fast",
23342334
status: "live",
@@ -2340,7 +2340,7 @@ describe.skip("InferenceClient", () => {
23402340
it(`textToImage - black-forest-labs/FLUX.1-schnell`, async () => {
23412341
const res = await client.textToImage({
23422342
model: "black-forest-labs/FLUX.1-schnell",
2343-
provider: "wavespeed-ai",
2343+
provider: "wavespeed",
23442344
inputs:
23452345
"Cute boy with a hat, exploring nature, holding a telescope, backpack, surrounded by flowers, cartoon style, vibrant colors.",
23462346
});
@@ -2350,7 +2350,7 @@ describe.skip("InferenceClient", () => {
23502350
it(`textToImage - openfree/flux-chatgpt-ghibli-lora`, async () => {
23512351
const res = await client.textToImage({
23522352
model: "openfree/flux-chatgpt-ghibli-lora",
2353-
provider: "wavespeed-ai",
2353+
provider: "wavespeed",
23542354
inputs:
23552355
"Cute boy with a hat, exploring nature, holding a telescope, backpack, surrounded by flowers, cartoon style, vibrant colors.",
23562356
});
@@ -2360,7 +2360,7 @@ describe.skip("InferenceClient", () => {
23602360
it(`textToImage - linoyts/yarn_art_Flux_LoRA`, async () => {
23612361
const res = await client.textToImage({
23622362
model: "linoyts/yarn_art_Flux_LoRA",
2363-
provider: "wavespeed-ai",
2363+
provider: "wavespeed",
23642364
inputs:
23652365
"Cute boy with a hat, exploring nature, holding a telescope, backpack, surrounded by flowers, cartoon style, vibrant colors.",
23662366
});
@@ -2370,7 +2370,7 @@ describe.skip("InferenceClient", () => {
23702370
it(`textToVideo - Wan-AI/Wan2.1-T2V-14B`, async () => {
23712371
const res = await client.textToVideo({
23722372
model: "Wan-AI/Wan2.1-T2V-14B",
2373-
provider: "wavespeed-ai",
2373+
provider: "wavespeed",
23742374
inputs:
23752375
"A cool street dancer, wearing a baggy hoodie and hip-hop pants, dancing in front of a graffiti wall, night neon background, quick camera cuts, urban trends.",
23762376
parameters: {
@@ -2389,7 +2389,7 @@ describe.skip("InferenceClient", () => {
23892389
it(`imageToImage - HiDream-ai/HiDream-E1-Full`, async () => {
23902390
const res = await client.imageToImage({
23912391
model: "HiDream-ai/HiDream-E1-Full",
2392-
provider: "wavespeed-ai",
2392+
provider: "wavespeed",
23932393
inputs: new Blob([readTestFile("cheetah.png")], { type: "image/png" }),
23942394
parameters: {
23952395
prompt: "The leopard chases its prey",

0 commit comments

Comments
 (0)