https://github.com/shavit/swift-bert

https://github.com/shavit/swift-bert

Last synced: 3 months ago
JSON representation

• Host: GitHub
• URL: https://github.com/shavit/swift-bert
• Owner: shavit
• Created: 2024-08-24T19:15:06.000Z (10 months ago)
• Default Branch: master
• Last Pushed: 2024-08-24T19:23:56.000Z (10 months ago)
• Last Synced: 2025-01-23T23:43:55.310Z (5 months ago)
• Language: Swift
• Size: 6.84 KB
• Stars: 0
• Watchers: 1
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        
# Swift BERT

> BERT implementation using BNNS

```

// From the dependency swift-transformers

let hub = HubApi(downloadBase: dest)

let repoName = "google/bert_uncased_L-2_H-128_A-2"

let modelDir = try await hub.snapshot(from: repoName)

let config = try hub.configuration(fileURL: modelDir.appending(path: "config.json"))

let weights = try Weights.from(fileURL: modelFile)

let model = BERTEmbedding(config: config, weights: weights)

let embeddings = try model(inputIDs: tokenIDs)

```