Spaces:
Sleeping
Sleeping
Update src/app/main_agent.py
Browse files- src/app/main_agent.py +2 -2
src/app/main_agent.py
CHANGED
@@ -59,8 +59,8 @@ import os
|
|
59 |
# Load model directly
|
60 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
61 |
|
62 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
63 |
-
model = AutoModelForCausalLM.from_pretrained("
|
64 |
|
65 |
def create_agent(accent_tool_obj) -> tuple[Runnable, Runnable]:
|
66 |
accent_tool = Tool(
|
|
|
59 |
# Load model directly
|
60 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
61 |
|
62 |
+
tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-3.1-8B")
|
63 |
+
model = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-3.1-8B")
|
64 |
|
65 |
def create_agent(accent_tool_obj) -> tuple[Runnable, Runnable]:
|
66 |
accent_tool = Tool(
|