159 lines
4.9 KiB
Swift
159 lines
4.9 KiB
Swift
// swift-tools-version: 6.1
|
|
// The swift-tools-version declares the minimum version of Swift required to build this package.
|
|
|
|
import CompilerPluginSupport
|
|
import PackageDescription
|
|
|
|
let package = Package(
|
|
name: "mlx-swift-lm",
|
|
platforms: [
|
|
.macOS(.v14),
|
|
.iOS(.v17),
|
|
.tvOS(.v17),
|
|
.visionOS(.v1),
|
|
],
|
|
products: [
|
|
.library(
|
|
name: "MLXLLM",
|
|
targets: ["MLXLLM"]),
|
|
.library(
|
|
name: "MLXVLM",
|
|
targets: ["MLXVLM"]),
|
|
.library(
|
|
name: "MLXLMCommon",
|
|
targets: ["MLXLMCommon"]),
|
|
.library(
|
|
name: "MLXEmbedders",
|
|
targets: ["MLXEmbedders"]),
|
|
.library(
|
|
name: "MLXHuggingFace",
|
|
targets: ["MLXHuggingFace"]),
|
|
.library(
|
|
name: "BenchmarkHelpers",
|
|
targets: ["BenchmarkHelpers"]),
|
|
.library(
|
|
name: "IntegrationTestHelpers",
|
|
targets: ["IntegrationTestHelpers"]),
|
|
],
|
|
dependencies: [
|
|
.package(url: "https://github.com/ml-explore/mlx-swift", .upToNextMinor(from: "0.31.3")),
|
|
.package(url: "https://github.com/swiftlang/swift-syntax.git", from: "600.0.0-latest"),
|
|
],
|
|
targets: [
|
|
.target(
|
|
name: "MLXLLM",
|
|
dependencies: [
|
|
"MLXLMCommon",
|
|
.product(name: "MLX", package: "mlx-swift"),
|
|
.product(name: "MLXNN", package: "mlx-swift"),
|
|
.product(name: "MLXOptimizers", package: "mlx-swift"),
|
|
],
|
|
path: "Libraries/MLXLLM",
|
|
exclude: [
|
|
"README.md"
|
|
]
|
|
),
|
|
.target(
|
|
name: "MLXVLM",
|
|
dependencies: [
|
|
"MLXLMCommon",
|
|
.product(name: "MLX", package: "mlx-swift"),
|
|
.product(name: "MLXNN", package: "mlx-swift"),
|
|
.product(name: "MLXOptimizers", package: "mlx-swift"),
|
|
],
|
|
path: "Libraries/MLXVLM",
|
|
exclude: [
|
|
"README.md"
|
|
]
|
|
),
|
|
.target(
|
|
name: "MLXLMCommon",
|
|
dependencies: [
|
|
.product(name: "MLX", package: "mlx-swift"),
|
|
.product(name: "MLXNN", package: "mlx-swift"),
|
|
.product(name: "MLXOptimizers", package: "mlx-swift"),
|
|
],
|
|
path: "Libraries/MLXLMCommon",
|
|
exclude: [
|
|
"README.md"
|
|
]
|
|
),
|
|
.target(
|
|
name: "MLXEmbedders",
|
|
dependencies: [
|
|
.product(name: "MLX", package: "mlx-swift"),
|
|
.product(name: "MLXNN", package: "mlx-swift"),
|
|
.target(name: "MLXLMCommon"),
|
|
],
|
|
path: "Libraries/MLXEmbedders",
|
|
exclude: [
|
|
"README.md"
|
|
]
|
|
),
|
|
.target(
|
|
name: "BenchmarkHelpers",
|
|
dependencies: [
|
|
"MLXLMCommon",
|
|
"MLXLLM",
|
|
"MLXVLM",
|
|
"MLXEmbedders",
|
|
.product(name: "MLX", package: "mlx-swift"),
|
|
],
|
|
path: "Libraries/BenchmarkHelpers"
|
|
),
|
|
.target(
|
|
name: "IntegrationTestHelpers",
|
|
dependencies: [
|
|
"MLXLMCommon",
|
|
"MLXLLM",
|
|
"MLXVLM",
|
|
"MLXEmbedders",
|
|
.product(name: "MLX", package: "mlx-swift"),
|
|
],
|
|
path: "Libraries/IntegrationTestHelpers",
|
|
exclude: ["README.md"]
|
|
),
|
|
.testTarget(
|
|
name: "MLXLMTests",
|
|
dependencies: [
|
|
.product(name: "MLX", package: "mlx-swift"),
|
|
.product(name: "MLXNN", package: "mlx-swift"),
|
|
.product(name: "MLXOptimizers", package: "mlx-swift"),
|
|
"MLXLMCommon",
|
|
"MLXLLM",
|
|
"MLXVLM",
|
|
"MLXEmbedders",
|
|
],
|
|
path: "Tests/MLXLMTests",
|
|
exclude: [
|
|
"README.md"
|
|
],
|
|
resources: [.process("Resources/1080p_30.mov"), .process("Resources/audio_only.mov")]
|
|
),
|
|
.macro(
|
|
name: "MLXHuggingFaceMacros",
|
|
dependencies: [
|
|
.product(name: "SwiftSyntaxMacros", package: "swift-syntax"),
|
|
.product(name: "SwiftCompilerPlugin", package: "swift-syntax"),
|
|
],
|
|
path: "Libraries/MLXHuggingFaceMacros"
|
|
),
|
|
.target(
|
|
name: "MLXHuggingFace",
|
|
dependencies: [
|
|
"MLXHuggingFaceMacros",
|
|
"MLXLMCommon",
|
|
],
|
|
path: "Libraries/MLXHuggingFace"
|
|
),
|
|
]
|
|
)
|
|
|
|
if Context.environment["MLX_SWIFT_BUILD_DOC"] == "1"
|
|
|| Context.environment["SPI_GENERATE_DOCS"] == "1"
|
|
{
|
|
package.dependencies.append(
|
|
.package(url: "https://github.com/apple/swift-docc-plugin", from: "1.3.0")
|
|
)
|
|
}
|