ash-171 commited on
Commit
74cfe7e
·
verified ·
1 Parent(s): 7bf0815

Update src/app/main_agent.py

Browse files
Files changed (1) hide show
  1. 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("google/gemma-3-1b-it")
63
- model = AutoModelForCausalLM.from_pretrained("google/gemma-3-1b-it")
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(