Allow loading a model using relative path (#64)

ngxson · Jun 8, 2024 · fd08e53 · fd08e53
1 parent 8e5edcb
commit fd08e53
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git 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
@@ -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>,