Fix #839 incorrect clip_timestamps being used in model (#842)

* Fix #839

Changed the code from updating the TranscriptionOptions class instead of the options object which likely was the cause of unexpected behaviour
This commit is contained in:
Natanael Tan
2024-05-17 17:35:07 +08:00
committed by GitHub
parent a1c3583c96
commit 4acdb5c619

View File

@@ -490,14 +490,16 @@ class WhisperModel:
content_duration = float(content_frames * self.feature_extractor.time_per_frame)
if isinstance(options.clip_timestamps, str):
TranscriptionOptions.clip_timestamps = [
float(ts)
for ts in (
options.clip_timestamps.split(",")
if options.clip_timestamps
else []
)
]
options = options._replace(
clip_timestamps=[
float(ts)
for ts in (
options.clip_timestamps.split(",")
if options.clip_timestamps
else []
)
]
)
seek_points: List[int] = [
round(ts * self.frames_per_second) for ts in options.clip_timestamps
]