SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix method name which changes in tutorial (#34252)

Browse Source 
The method `model_download_tool` was called `model_download_counter` earlier in the tutorial, this raises an error when following the code.
This commit is contained in:
Andrés Marafioti
2024-10-21 14:21:52 -03:00
committed by GitHub
parent f701b98e4a
commit 32590b5ecb
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
docs/source/en/agents.md
View File

@@ -332,7 +332,7 @@ This code can quickly be converted into a tool, just by wrapping it in a functio
from transformers import tool from transformers import tool
@tool @tool
def model_download_counter(task: str) -> str: def model_download_tool(task: str) -> str:
""" """
This is a tool that returns the most downloaded model of a given task on the Hugging Face Hub. This is a tool that returns the most downloaded model of a given task on the Hugging Face Hub.
It returns the name of the checkpoint. It returns the name of the checkpoint.
@@ -345,7 +345,7 @@ def model_download_counter(task: str) -> str:
``` ```
The function needs: The function needs:
- A clear name. The name usually describes what the tool does. Since the code returns the model with the most downloads for a task, let's put `model_download_counter`. - A clear name. The name usually describes what the tool does. Since the code returns the model with the most downloads for a task, let's put `model_download_tool`.
- Type hints on both inputs and output - Type hints on both inputs and output
- A description, that includes an 'Args:' part where each argument is described (without a type indication this time, it will be pulled from the type hint). - A description, that includes an 'Args:' part where each argument is described (without a type indication this time, it will be pulled from the type hint).
All these will be automatically baked into the agent's system prompt upon initialization: so strive to make them as clear as possible! All these will be automatically baked into the agent's system prompt upon initialization: so strive to make them as clear as possible!
@@ -367,7 +367,7 @@ You get the following:
======== New task ======== ======== New task ========
Can you give me the name of the model that has the most downloads in the 'text-to-video' task on the Hugging Face Hub? Can you give me the name of the model that has the most downloads in the 'text-to-video' task on the Hugging Face Hub?
==== Agent is executing the code below: ==== Agent is executing the code below:
most_downloaded_model = model_download_counter(task="text-to-video") most_downloaded_model = model_download_tool(task="text-to-video")
print(f"The most downloaded model for the 'text-to-video' task is {most_downloaded_model}.") print(f"The most downloaded model for the 'text-to-video' task is {most_downloaded_model}.")
==== ====
``` ```