Skip to content

Commit 69de809

Browse files
authored
Consolidate TensorUtils into Generation and reorganize targets (#244)
* Remove TensorUtils dependency from Models target * Consolidate TensorUtils into Generation and reorganize targets
1 parent 07f32d9 commit 69de809

20 files changed

+10
-11
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),
34+
.target(name: "Models", dependencies: ["Tokenizers", "Generation"]),
3335
.target(name: "Tokenizers", dependencies: ["Hub", .product(name: "Jinja", package: "Jinja")]),
34-
.target(name: "TensorUtils"),
35-
.target(name: "Generation", dependencies: ["Tokenizers", "TensorUtils"]),
36-
.target(name: "Models", dependencies: ["Tokenizers", "Generation", "TensorUtils"]),
37-
.testTarget(name: "TokenizersTests", dependencies: ["Tokenizers", "Models", "Hub"], resources: [.process("Resources"), .process("Vocabs")]),
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)