From fd08e535e889538b604d597355c39b388ab1e33b Mon Sep 17 00:00:00 2001
From: Victor Nogueira <felladrin@gmail.com>
Date: Sat, 8 Jun 2024 16:11:43 +0300
Subject: [PATCH] Allow loading a model using relative path (#64)

---
 src/downloader/multi-downloads.ts | 2 +-
 src/downloader/remote-blob.ts     | 6 +++---
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/downloader/multi-downloads.ts b/src/downloader/multi-downloads.ts
index 58cf0e3..a27edfe 100644
--- a/src/downloader/multi-downloads.ts
+++ b/src/downloader/multi-downloads.ts
@@ -45,7 +45,7 @@ export class MultiDownloads {
   async run(): Promise<Blob[]> {
     // create all Blobs
     await Promise.all(this.tasks.map(async (task) => {
-      task.blob = await GGUFRemoteBlob.create(new URL(task.url), {
+      task.blob = await GGUFRemoteBlob.create(task.url, {
         logger: this.logger,
         useCache: this.useCache,
         startSignal: task.signalStart,
diff --git a/src/downloader/remote-blob.ts b/src/downloader/remote-blob.ts
index 2323ac8..2142108 100644
--- a/src/downloader/remote-blob.ts
+++ b/src/downloader/remote-blob.ts
@@ -25,7 +25,7 @@ interface GGUFRemoteBlobCreateOptions {
 }
 
 export class GGUFRemoteBlob extends Blob {
-  static async create(url: URL, opts?: GGUFRemoteBlobCreateOptions): Promise<Blob> {
+  static async create(url: RequestInfo | URL, opts?: GGUFRemoteBlobCreateOptions): Promise<Blob> {
     const customFetch = opts?.fetch ?? fetch;
     const response = await customFetch(url, { method: 'HEAD' });
 
@@ -62,7 +62,7 @@ export class GGUFRemoteBlob extends Blob {
     }
   }
 
-  private url: URL;
+  private url: RequestInfo | URL;
   private start: number;
   private end: number;
   private contentType: string;
@@ -72,7 +72,7 @@ export class GGUFRemoteBlob extends Blob {
   private progressCallback: ProgressCallback;
   private startSignal?: Promise<void>;
 
-  constructor(url: URL, start: number, end: number, contentType: string, full: boolean, customFetch: typeof fetch, additionals: {
+  constructor(url: RequestInfo | URL, start: number, end: number, contentType: string, full: boolean, customFetch: typeof fetch, additionals: {
     cachedStream?: ReadableStream,
     progressCallback: ProgressCallback,
     startSignal?: Promise<void>,