fix(core): fix Model.example_output for embedding models

pyannote · Sep 16, 2023 · 9df6944 · 9df6944
1 parent b660b1e
commit 9df6944
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/pyannote/audio/core/model.py b/pyannote/audio/core/model.py
@@ -198,12 +198,13 @@ def __example_output(
             example_output: torch.Tensor,
             specifications: Specifications = None,
         ) -> Output:
-            _, num_frames, dimension = example_output.shape
-
             if specifications.resolution == Resolution.FRAME:
+                _, num_frames, dimension = example_output.shape
                 frame_duration = specifications.duration / num_frames
                 frames = SlidingWindow(step=frame_duration, duration=frame_duration)
             else:
+                _, dimension = example_output.shape
+                num_frames = None
                 frames = None
 
             return Output(