diff --git a/Sources/FluidAudio/Diarizer/Offline/Core/OfflineDiarizerModels.swift b/Sources/FluidAudio/Diarizer/Offline/Core/OfflineDiarizerModels.swift index 980fddb63..cfe5cb5a4 100644 --- a/Sources/FluidAudio/Diarizer/Offline/Core/OfflineDiarizerModels.swift +++ b/Sources/FluidAudio/Diarizer/Offline/Core/OfflineDiarizerModels.swift @@ -17,6 +17,7 @@ public struct OfflineDiarizerModels: Sendable { private static func loadPLDAPsi(from directory: URL) throws -> [Double] { let candidatePaths = [ directory.appendingPathComponent("plda-parameters.json", isDirectory: false), + directory.appendingPathComponent("speaker-diarization/plda-parameters.json", isDirectory: false), directory.appendingPathComponent("speaker-diarization-coreml/plda-parameters.json", isDirectory: false), directory.appendingPathComponent("speaker-diarization-offline/plda-parameters.json", isDirectory: false), ] diff --git a/Sources/FluidAudio/ModelNames.swift b/Sources/FluidAudio/ModelNames.swift index bb52aa574..5f3ff3f01 100644 --- a/Sources/FluidAudio/ModelNames.swift +++ b/Sources/FluidAudio/ModelNames.swift @@ -129,16 +129,8 @@ public enum Repo: String, CaseIterable { return "nemotron-streaming/560ms" case .sortformer: return "sortformer" - case .lseend: - return "ls-eend" - case .pocketTts: - return "pocket-tts" - case .multilingualG2p: - return "charsiu-g2p-byt5" - case .parakeetTdtCtc110m: - return "parakeet-tdt-ctc-110m" default: - return name + return name.replacingOccurrences(of: "-coreml", with: "") } } } @@ -166,6 +158,7 @@ public enum ModelNames { public static let fbank = "FBank" public static let embedding = "Embedding" public static let pldaRho = "PldaRho" + public static let pldaParameters = "plda-parameters.json" public static let segmentationFile = segmentation + ".mlmodelc" public static let fbankFile = fbank + ".mlmodelc" @@ -182,6 +175,7 @@ public enum ModelNames { fbankPath, embeddingPath, pldaRhoPath, + pldaParameters, ] } diff --git a/Tests/FluidAudioTests/ASR/Parakeet/ModelNamesTests.swift b/Tests/FluidAudioTests/ASR/Parakeet/ModelNamesTests.swift index 3e3607394..6048ff95c 100644 --- a/Tests/FluidAudioTests/ASR/Parakeet/ModelNamesTests.swift +++ b/Tests/FluidAudioTests/ASR/Parakeet/ModelNamesTests.swift @@ -125,7 +125,7 @@ final class ModelNamesTests: XCTestCase { // Verify name (repo slug with -coreml suffix) XCTAssertEqual(repo.name, "parakeet-tdt-ctc-110m-coreml") - // Verify folder name (simplified local folder name) + // Verify folder name (simplified - strips -coreml suffix by default) XCTAssertEqual(repo.folderName, "parakeet-tdt-ctc-110m") // Should have no subpath (not a variant repo)