From 7e1f8ca3053ce2c5c242a1d755d248e032ca23c6 Mon Sep 17 00:00:00 2001 From: Pedro Cuenca Date: Wed, 4 Sep 2024 11:57:25 +0200 Subject: [PATCH 1/3] Fix WBlob test --- packages/hub/src/utils/WebBlob.spec.ts | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/packages/hub/src/utils/WebBlob.spec.ts b/packages/hub/src/utils/WebBlob.spec.ts index 75687ae1fa..bb837a5a5f 100644 --- a/packages/hub/src/utils/WebBlob.spec.ts +++ b/packages/hub/src/utils/WebBlob.spec.ts @@ -51,15 +51,14 @@ describe("WebBlob", () => { it("should lazy load a LFS file hosted on Hugging Face", async () => { const stableDiffusionUrl = - "https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/39593d5650112b4cc580433f6b0435385882d819/v1-5-pruned.safetensors"; + "https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0/resolve/main/unet/diffusion_pytorch_model.fp16.safetensors"; const url = new URL(stableDiffusionUrl); const webBlob = await WebBlob.create(url); - expect(webBlob.size).toBe(7_703_324_286); + expect(webBlob.size).toBe(5_135_149_760); expect(webBlob).toBeInstanceOf(WebBlob); expect(webBlob).toMatchObject({ url }); - expect(base64FromBytes(new Uint8Array(await webBlob.slice(6, 12).arrayBuffer()))).toBe("AAB7Il9f"); - expect(base64FromBytes(new Uint8Array(await webBlob.slice(0, 12).arrayBuffer()))).toBe("ytIDAAAAAAB7Il9f"); + expect((await webBlob.slice(10, 22).text())).toBe("__metadata__"); }); it("should create a slice on the file", async () => { From fb217d7e521c9e3b2f344420a7e66da1bf123155 Mon Sep 17 00:00:00 2001 From: Pedro Cuenca Date: Wed, 4 Sep 2024 11:59:39 +0200 Subject: [PATCH 2/3] Not using base64 for the test --- packages/hub/src/utils/WebBlob.spec.ts | 1 - 1 file changed, 1 deletion(-) diff --git a/packages/hub/src/utils/WebBlob.spec.ts b/packages/hub/src/utils/WebBlob.spec.ts index bb837a5a5f..a9ed43d9db 100644 --- a/packages/hub/src/utils/WebBlob.spec.ts +++ b/packages/hub/src/utils/WebBlob.spec.ts @@ -1,6 +1,5 @@ import { describe, expect, it, beforeAll } from "vitest"; import { WebBlob } from "./WebBlob"; -import { base64FromBytes } from "./base64FromBytes"; describe("WebBlob", () => { const resourceUrl = new URL("https://huggingface.co/spaces/aschen/push-model-from-web/raw/main/mobilenet/model.json"); From c26ec52f493fec960b9f9725702c7c09a526460b Mon Sep 17 00:00:00 2001 From: Pedro Cuenca Date: Wed, 4 Sep 2024 12:01:59 +0200 Subject: [PATCH 3/3] Remove spurious parenthesis --- packages/hub/src/utils/WebBlob.spec.ts | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/packages/hub/src/utils/WebBlob.spec.ts b/packages/hub/src/utils/WebBlob.spec.ts index a9ed43d9db..919c8f0199 100644 --- a/packages/hub/src/utils/WebBlob.spec.ts +++ b/packages/hub/src/utils/WebBlob.spec.ts @@ -57,7 +57,7 @@ describe("WebBlob", () => { expect(webBlob.size).toBe(5_135_149_760); expect(webBlob).toBeInstanceOf(WebBlob); expect(webBlob).toMatchObject({ url }); - expect((await webBlob.slice(10, 22).text())).toBe("__metadata__"); + expect(await webBlob.slice(10, 22).text()).toBe("__metadata__"); }); it("should create a slice on the file", async () => {