From 69ed767b2f1a6e671a9de8d55a63ee9cdd1174d0 Mon Sep 17 00:00:00 2001 From: coyotte508 Date: Wed, 30 Oct 2024 12:42:57 +0100 Subject: [PATCH 1/2] fix-model-info --- packages/hub/src/lib/model-info.spec.ts | 2 ++ packages/hub/src/lib/model-info.ts | 2 +- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/packages/hub/src/lib/model-info.spec.ts b/packages/hub/src/lib/model-info.spec.ts index cdbad71c16..8767ca18a6 100644 --- a/packages/hub/src/lib/model-info.spec.ts +++ b/packages/hub/src/lib/model-info.spec.ts @@ -5,10 +5,12 @@ describe("modelInfo", () => { it("should return the model info", async () => { const info = await modelInfo({ name: "openai-community/gpt2", + additionalFields: ["author"], }); expect(info).toEqual({ id: "621ffdc036468d709f17434d", downloads: expect.any(Number), + author: "OpenAI", gated: false, name: "openai-community/gpt2", updatedAt: expect.any(Date), diff --git a/packages/hub/src/lib/model-info.ts b/packages/hub/src/lib/model-info.ts index 2744905c05..02f41d717e 100644 --- a/packages/hub/src/lib/model-info.ts +++ b/packages/hub/src/lib/model-info.ts @@ -7,7 +7,7 @@ import { pick } from "../utils/pick"; import { MODEL_EXPAND_KEYS, type MODEL_EXPANDABLE_KEYS, type ModelEntry } from "./list-models"; export async function modelInfo< - const T extends Exclude<(typeof MODEL_EXPANDABLE_KEYS)[number], (typeof MODEL_EXPANDABLE_KEYS)[number]> = never, + const T extends Exclude<(typeof MODEL_EXPANDABLE_KEYS)[number], (typeof MODEL_EXPAND_KEYS)[number]> = never, >( params: { name: string; From f5a3c8d1e815c33e0080ce490f5244607b4f6221 Mon Sep 17 00:00:00 2001 From: coyotte508 Date: Wed, 30 Oct 2024 12:45:09 +0100 Subject: [PATCH 2/2] fixup! fix-model-info --- packages/hub/src/lib/model-info.spec.ts | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/packages/hub/src/lib/model-info.spec.ts b/packages/hub/src/lib/model-info.spec.ts index 8767ca18a6..81e6f82467 100644 --- a/packages/hub/src/lib/model-info.spec.ts +++ b/packages/hub/src/lib/model-info.spec.ts @@ -10,7 +10,7 @@ describe("modelInfo", () => { expect(info).toEqual({ id: "621ffdc036468d709f17434d", downloads: expect.any(Number), - author: "OpenAI", + author: "openai-community", gated: false, name: "openai-community/gpt2", updatedAt: expect.any(Date),