Skip to content

Commit 4a856a9

Browse files
committed
Consolidate TensorUtils into Generation and reorganize targets
1 parent e70d791 commit 4a856a9

20 files changed

+9
-10
lines changed

Package.swift

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -29,13 +29,13 @@ let package = Package(
2929
]
3030
),
3131
.executableTarget(name: "HubCLI", dependencies: ["Hub", .product(name: "ArgumentParser", package: "swift-argument-parser")]),
32+
.target(name: "Generation", dependencies: ["Tokenizers"]),
3233
.target(name: "Hub", resources: [.process("FallbackConfigs")], swiftSettings: swiftSettings),
33-
.target(name: "Tokenizers", dependencies: ["Hub", .product(name: "Jinja", package: "Jinja")]),
34-
.target(name: "TensorUtils"),
35-
.target(name: "Generation", dependencies: ["Tokenizers", "TensorUtils"]),
3634
.target(name: "Models", dependencies: ["Tokenizers", "Generation"]),
37-
.testTarget(name: "TokenizersTests", dependencies: ["Tokenizers", "Models", "Hub"], resources: [.process("Resources"), .process("Vocabs")]),
35+
.target(name: "Tokenizers", dependencies: ["Hub", .product(name: "Jinja", package: "Jinja")]),
36+
.testTarget(name: "GenerationTests", dependencies: ["Generation"]),
3837
.testTarget(name: "HubTests", dependencies: ["Hub", .product(name: "Jinja", package: "Jinja")], swiftSettings: swiftSettings),
39-
.testTarget(name: "TensorUtilsTests", dependencies: ["TensorUtils", "Models", "Hub"], resources: [.process("Resources")]),
38+
.testTarget(name: "ModelsTests", dependencies: ["Models", "Hub"], resources: [.process("Resources")]),
39+
.testTarget(name: "TokenizersTests", dependencies: ["Tokenizers", "Models", "Hub"], resources: [.process("Resources"), .process("Vocabs")]),
4040
]
4141
)

Sources/Generation/Generation.swift

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,6 @@
66
//
77

88
import CoreML
9-
import TensorUtils
109
import Tokenizers
1110

1211
public enum GenerationMode {

0 commit comments

Comments
 (0)