squeeze-ai-lab / TinyAgent-ToolRAG

huggingface.co
Total runs: 6
24-hour runs: 0
7-day runs: -1
30-day runs: -987
Model's Last Updated: May 30 2024
text-classification

Introduction of TinyAgent-ToolRAG

Model Details of TinyAgent-ToolRAG

TinyAgent: Function Calling at the Edge

Get the desktop app ‎ ‎ | Read the blog post

Thumbnail

TinyAgent aims to enable complex reasoning and function calling capabilities in Small Language Models (SLMs) that can be deployed securely and privately at the edge. Traditional Large Language Models (LLMs) like GPT-4 and Gemini-1.5, while powerful, are often too large and resource-intensive for edge deployment, posing challenges in terms of privacy, connectivity, and latency. TinyAgent addresses these challenges by training specialized SLMs with high-quality, curated data, and focusing on function calling with LLMCompiler . As a driving application, TinyAgent can interact with various MacOS applications, assisting users with day-to-day tasks such as composing emails, managing contacts, scheduling calendar events, and organizing Zoom meetings.

When faced with challenging tasks, SLM agents require appropriate tools and in-context examples to guide them. If the model sees irrelevant examples, it can hallucinate. Likewise, if the model sees the descriptions of the tools that it doesn’t need, it usually gets confused, and these tools take up unnecessary prompt space. To tackle this, TinyAgent uses ToolRAG to retrieve the best tools and examples suited for a given query. This process has minimal latency and increases the accuracy of TinyAgent substantially. Please take a look at our blog post for more details.

Model Developers: Squeeze AI Lab at University of California, Berkeley.

Variations: TinyAgent models come in 2 sizes: TinyAgent-1.1B and TinyAgent-7B

License: MIT

Demo
TinyAgent Demo
How to Use

Please see our Github for details on how to use TinyAgent models. TinyAgent models can be used programmatically or through our user interface.

Training Details

Dataset: We curated a dataset of 40,000 real-life use cases. We use GPT-3.5-Turbo to generate real-world instructions. These are then used to obtain synthetic execution plans using GPT-4-Turbo. Please check out our blog post for more details on our dataset.

Fine-tuning Procedure: TinyAgent models are fine-tuned from base models. Below is a table of each TinyAgent model with its base counterpart

Model Success Rate
GPT-3.5-turbo 65.04%
GPT-4-turbo 79.08%
TinyLLama-1.1B-32K-Instruct 12.71%
WizardLM-2-7b 41.25%
TinyAgent-1.1B + ToolRAG / [ hf ] [ gguf ] 80.06%
TinyAgent-7B + ToolRAG / [ hf ] [ gguf ] 84.95%

Using the synthetic data generation process described above, we use parameter-efficient fine-tuning with LoRA to fine-tune the base models for 3 epochs. Please check out our blog post for more details on our fine-tuning procedure.

Links

Blog Post : https://bair.berkeley.edu/blog/2024/05/29/tiny-agent/

Github: https://github.com/SqueezeAILab/TinyAgent

Runs of squeeze-ai-lab TinyAgent-ToolRAG on huggingface.co

6
Total runs
0
24-hour runs
0
3-day runs
-1
7-day runs
-987
30-day runs

More Information About TinyAgent-ToolRAG huggingface.co Model

TinyAgent-ToolRAG huggingface.co

TinyAgent-ToolRAG huggingface.co is an AI model on huggingface.co that provides TinyAgent-ToolRAG's model effect (), which can be used instantly with this squeeze-ai-lab TinyAgent-ToolRAG model. huggingface.co supports a free trial of the TinyAgent-ToolRAG model, and also provides paid use of the TinyAgent-ToolRAG. Support call TinyAgent-ToolRAG model through api, including Node.js, Python, http.

squeeze-ai-lab TinyAgent-ToolRAG online free

TinyAgent-ToolRAG huggingface.co is an online trial and call api platform, which integrates TinyAgent-ToolRAG's modeling effects, including api services, and provides a free online trial of TinyAgent-ToolRAG, you can try TinyAgent-ToolRAG online for free by clicking the link below.

squeeze-ai-lab TinyAgent-ToolRAG online free url in huggingface.co:

https://huggingface.co/squeeze-ai-lab/TinyAgent-ToolRAG

TinyAgent-ToolRAG install

TinyAgent-ToolRAG is an open source model from GitHub that offers a free installation service, and any user can find TinyAgent-ToolRAG on GitHub to install. At the same time, huggingface.co provides the effect of TinyAgent-ToolRAG install, users can directly use TinyAgent-ToolRAG installed effect in huggingface.co for debugging and trial. It also supports api for free installation.

TinyAgent-ToolRAG install url in huggingface.co:

https://huggingface.co/squeeze-ai-lab/TinyAgent-ToolRAG

Url of TinyAgent-ToolRAG

Provider of TinyAgent-ToolRAG huggingface.co

squeeze-ai-lab
ORGANIZATIONS

Other API from squeeze-ai-lab