Update torchvision

This commit is contained in:
Aymeric 2024-12-27 17:19:07 +01:00
parent a9bcfbdd22
commit 132f9f3e6a
3 changed files with 3 additions and 2 deletions

View File

@ -66,7 +66,7 @@ Especially, since code execution can be a security concern (arbitrary code execu
## How smol is it really?
We strived to keep abstractions to a strict minimum: the main code in `agents.py` is only ~1,000 lines of code.
Still, we implement several types of agents: `CodeAgent` writes its actions as Python code snippets, and the more classic `ToolCallingAgent` that leverages built-in tool calling methods.
Still, we implement several types of agents: `CodeAgent` writes its actions as Python code snippets, and the more classic `ToolCallingAgent` leverages built-in tool calling methods.
By the way, why use a framework at all? Well, because a big part of this stuff is non-trivial. For instance, the code agent has to keep a consistent format for code throughout its system prompt, its parser, the execution. So our framework handles this complexity for you. But of course we still encourage you to hack into the source code and use only the bits that you need, to the exclusion of everything else!

View File

@ -26,6 +26,7 @@ dependencies = [
"e2b-code-interpreter>=1.0.3",
"torch>=2.5.1",
"litellm>=1.55.10",
"torchvision>=0.20.1",
]
[project.optional-dependencies]

View File

@ -219,7 +219,7 @@ class HfApiModel(Model):
def __init__(
self,
model_id: str = "meta-llama/Llama-3.3-70B-Instruct",
model_id: str = "Qwen/Qwen2.5-Coder-32B-Instruct",
token: Optional[str] = None,
timeout: Optional[int] = 120,
):