diff --git a/Transformer/PythonCheckpointReader.swift b/Transformer/PythonCheckpointReader.swift index 2a175c10f36..d46815d320e 100644 --- a/Transformer/PythonCheckpointReader.swift +++ b/Transformer/PythonCheckpointReader.swift @@ -14,7 +14,7 @@ import TensorFlow -struct Config : Codable { +struct Config: Codable { let vocabSize: Int let contextSize: Int let embeddingSize: Int diff --git a/Transformer/main.swift b/Transformer/main.swift index 46a6aa5b7e6..0cc91eb2171 100644 --- a/Transformer/main.swift +++ b/Transformer/main.swift @@ -23,7 +23,7 @@ let encoder = Python.import("encoder").get_encoder(modelName) let checkpoint = "models/\(modelName)/model.ckpt" let configFile = "models/\(modelName)/hparams.json" -let configData = try Data.init(contentsOf: URL(fileURLWithPath: configFile)) +let configData = try Data(contentsOf: URL(fileURLWithPath: configFile)) let config = try JSONDecoder().decode(Config.self, from: configData) let model = TransformerLM( contentsOfPythonCheckpointFile: checkpoint, config: config, scope: "model")