@omote/core 0.4.1 → 0.4.2

package/dist/index.js

@@ -3092,12 +3092,23 @@ var Wav2Vec2Inference = class {
       logger2.debug("Running warmup inference to initialize GPU context");
       const warmupStart = performance.now();
       const silentAudio = new Float32Array(16e3);
-      await this.infer(silentAudio, 0);
+      const WARMUP_TIMEOUT_MS = 15e3;
+      const warmupResult = await Promise.race([
+        this.infer(silentAudio, 0).then(() => "ok"),
+        new Promise((r) => setTimeout(() => r("timeout"), WARMUP_TIMEOUT_MS))
+      ]);
       const warmupTimeMs = performance.now() - warmupStart;
-      logger2.info("Warmup inference complete", {
-        warmupTimeMs: Math.round(warmupTimeMs),
-        backend: this._backend
-      });
+      if (warmupResult === "timeout") {
+        logger2.warn("Warmup inference timed out \u2014 GPU may be unresponsive. Continuing without warmup.", {
+          timeoutMs: WARMUP_TIMEOUT_MS,
+          backend: this._backend
+        });
+      } else {
+        logger2.info("Warmup inference complete", {
+          warmupTimeMs: Math.round(warmupTimeMs),
+          backend: this._backend
+        });
+      }
       telemetry?.recordHistogram("omote.model.warmup_time", warmupTimeMs, {
         model: "wav2vec2",
         backend: this._backend
@@ -4153,20 +4164,21 @@ var LipSyncWithFallback = class {
     try {
       return await this.implementation.load();
     } catch (error) {
-      logger6.warn("GPU model load failed, falling back to CPU model", {
-        error: error instanceof Error ? error.message : String(error)
-      });
-      try {
-        await this.implementation.dispose();
-      } catch {
-      }
-      this.implementation = new Wav2ArkitCpuInference({
-        modelUrl: this.config.cpuModelUrl
-      });
-      this.hasFallenBack = true;
-      logger6.info("Fallback to Wav2ArkitCpuInference successful");
-      return await this.implementation.load();
+      return this.fallbackToCpu(error instanceof Error ? error.message : String(error));
+    }
+  }
+  async fallbackToCpu(reason) {
+    logger6.warn("GPU model load failed, falling back to CPU model", { reason });
+    try {
+      await this.implementation.dispose();
+    } catch {
     }
+    this.implementation = new Wav2ArkitCpuInference({
+      modelUrl: this.config.cpuModelUrl
+    });
+    this.hasFallenBack = true;
+    logger6.info("Fallback to Wav2ArkitCpuInference successful");
+    return await this.implementation.load();
   }
   async infer(audioSamples, identityIndex) {
     return this.implementation.infer(audioSamples, identityIndex);