Fix code docs that got out of sync (#272)

* Fix doc for `callAsFunction` that got out of sync

* More fixes

Author: pcuenca

Examples/transformers-cli/Sources/transformers-cli/Transformers.swift

@@ -125,7 +125,7 @@ enum ComputeUnits: String, ExpressibleByArgument, CaseIterable {
 
 /// Returns a cleaned and formatted version of the response.
 ///
-/// - Parameter respone: The response to clean and format.
+/// - Parameter response: The response to clean and format.
 /// - Returns: A 'user friendly' representation of the generated response.
 private func formatResponse(_ response: String) -> String {
     response

Sources/Models/LanguageModel.swift

@@ -236,8 +236,7 @@ public extension LanguageModel {
 
     /// Determines the type of KV Cache available for the model, if any.
     ///
-    /// - Parameters:
-    ///   - model: The Core ML model
+    /// - Parameter model: The Core ML model
     /// - Returns: The type of KV Cache available.
     fileprivate static func kvCacheAvailability(for model: MLModel) -> KVCacheAvailability? {
         func isStatefulKVCacheAvailable(for model: MLModel) -> Bool {

Sources/Models/LanguageModelTypes.swift

@@ -57,9 +57,9 @@ public extension LanguageModelProtocol {
     /// This provides a more convenient syntax for calling `predictNextTokenScores`.
     ///
     /// - Parameters:
-    ///   - tokens: The input token sequence
-    ///   - config: The generation configuration containing model parameters
-    /// - Returns: A shaped array containing the logits for the next token prediction
+    ///   - input: The input sequence tensor.
+    ///   - config: The generation configuration containing model parameters.
+    /// - Returns: MLTensor with the raw scores of the next token.
     func callAsFunction(_ input: MLTensor, config: GenerationConfig) async -> MLTensor {
         await predictNextTokenScores(input, config: config)
     }