Files
mlx-swift-examples/Tools/llm-tool
2024-02-22 12:46:44 -08:00
..
2024-02-22 10:41:02 -08:00

llm-tool

See various READMEs:

Building

Build the llm-tool scheme in Xcode.

Running (Xcode)

To run this in Xcode simply press cmd-opt-r to set the scheme arguments. For example:

--model mlx-community/Mistral-7B-v0.1-hf-4bit-mlx
--prompt "swift programming language"
--max-tokens 50

Then cmd-r to run.

Note: you may be prompted for access to your Documents directory -- this is where the Hugging Face HubApi stores the downloaded files.

The model should be a path in the Hugging Face repository, e.g.:

  • mlx-community/Mistral-7B-v0.1-hf-4bit-mlx
  • mlx-community/phi-2-hf-4bit-mlx

See LLM for more info.

Running (Command Line)

llm-tool can also be run from the command line if built from Xcode, but the DYLD_FRAMEWORK_PATH must be set so that the frameworks and bundles can be found:

The easiest way to do this is drag the Products/llm-tool into Terminal to get the path:

DYLD_FRAMEWORK_PATH=~/Library/Developer/Xcode/DerivedData/mlx-examples-swift-ceuohnhzsownvsbbleukxoksddja/Build/Products/Debug ~/Library/Developer/Xcode/DerivedData/mlx-examples-swift-ceuohnhzsownvsbbleukxoksddja/Build/Products/Debug/llm-tool --prompt "swift programming language"