David Koski
0374e4b073
update documentation
2024-03-01 16:33:49 -08:00
David Koski
807c8136c0
update README
2024-03-01 16:13:29 -08:00
David Koski
c49dd73c28
swift-format, circleci setup
2024-03-01 16:10:34 -08:00
David Koski
b41f14fba7
add LLM evaluator example
...
- runs on iOS and macOS
- downloads a model / tokenizer from hugging face
- evaluates the given prompt
2024-03-01 16:10:00 -08:00
David Koski
79e0620891
add MNIST training example
2024-03-01 15:55:36 -08:00
David Koski
2157333905
swift-format!
2024-03-01 14:47:43 -08:00
David Koski
82f6a969d4
llm improvements
...
- document the tokenizer used (https://github.com/huggingface/swift-transformers )
- provide a hook for tokenizer configuration, prompt augmentation
- this isn't as rich as the python equivalents but it helps a little
2024-03-01 14:46:32 -08:00
David Koski
599661774a
Merge pull request #6 from rudro/rudro_fixmain
...
Fix position of @main annotation
2024-02-27 13:32:39 -08:00
Rudro Samanta
362f1c8ad8
Fix position of @main annotation
...
- Also update package versions otherwise things don't compile out of the box
(you need the version where `callAsFunction` is marked `open`)
2024-02-27 13:02:13 -08:00
David Koski
3f02fcc1cb
expose eosToken
2024-02-26 14:58:51 -08:00
David Koski
4fad86d84b
split tokenizer code out into new file
2024-02-26 14:42:40 -08:00
David Koski
c7919cf7fe
fix rmsnorm for gemma
2024-02-26 14:09:48 -08:00
David Koski
a2ff291608
add reference to filed issue
2024-02-26 13:31:33 -08:00
David Koski
c86d1c195e
partial fix for #1
...
- handle loading models with different names for the safetensors files (gemma)
- handle merge tokens that can't be split
- organize code into Load/Evaluate
2024-02-26 13:23:21 -08:00
David Koski
d666271ede
try to refresh packages
2024-02-26 11:57:29 -08:00
David Koski
bb7bacc077
fix for #2 -- CodeLlama crashes
...
- add replacement tokenizer class for unknown tokenizers
- fix quantization for models that don't have lm_head quantized
Requires https://github.com/ml-explore/mlx-swift/pull/28
2024-02-26 10:38:05 -08:00
David Koski
8870b0d386
add notes about building Release
2024-02-26 10:34:59 -08:00
David Koski
5a83d7d92b
fix broken links, clarify documentation
2024-02-22 12:46:44 -08:00
David Koski
c917f5b524
add admin files
2024-02-22 12:33:18 -08:00
David Koski
a273fa1777
alllow build plugins to run
2024-02-22 12:20:23 -08:00
David Koski
b6d1e14465
initial commit
2024-02-22 10:41:02 -08:00
