Enable passing extra args to load function in classes dervied from BaseModule

Author: pweglik

packages/react-native-executorch/src/modules/BaseModule.ts

@@ -7,13 +7,16 @@ export class BaseModule {
   static onDownloadProgressCallback: (downloadProgress: number) => void =
     () => {};
 
-  static async load(...sources: ResourceSource[]): Promise<void> {
+  static async load(
+    sources: ResourceSource[],
+    ...loadArgs: any[] // this can be used in derived classes to pass extra args to load method
+  ): Promise<void> {
     try {
       const paths = await ResourceFetcher.fetchMultipleResources(
         this.onDownloadProgressCallback,
         ...sources
       );
-      await this.nativeModule.loadModule(...paths);
+      await this.nativeModule.loadModule(...paths, ...loadArgs);
     } catch (error) {
       throw new Error(getError(error));
     }

packages/react-native-executorch/src/modules/general/ExecutorchModule.ts

@@ -9,7 +9,7 @@ export class ExecutorchModule extends BaseModule {
   protected static override nativeModule = ETModuleNativeModule;
 
   static override async load(modelSource: ResourceSource) {
-    return await super.load(modelSource);
+    return await super.load([modelSource]);
   }
 
   static override async forward(input: ETInput[] | ETInput, shape: number[][]) {

packages/react-native-executorch/src/modules/natural_language_processing/TextEmbeddingsModule.ts

@@ -9,7 +9,7 @@ export class TextEmbeddingsModule extends BaseModule {
     modelSource: ResourceSource,
     tokenizerSource: ResourceSource
   ) {
-    await super.load(modelSource, tokenizerSource);
+    await super.load([modelSource, tokenizerSource]);
   }
 
   static override async forward(input: string): Promise<number[]> {

packages/react-native-executorch/src/modules/natural_language_processing/TokenizerModule.ts

@@ -6,7 +6,7 @@ export class TokenizerModule extends BaseModule {
   protected static override nativeModule = TokenizerNativeModule;
 
   static override async load(tokenizerSource: ResourceSource) {
-    await super.load(tokenizerSource);
+    await super.load([tokenizerSource]);
   }
 
   static async decode(