diff options
Diffstat (limited to 'collaborativeagents/agents/local_user_agent.py')
| -rw-r--r-- | collaborativeagents/agents/local_user_agent.py | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/collaborativeagents/agents/local_user_agent.py b/collaborativeagents/agents/local_user_agent.py index eae311e..fd085c4 100644 --- a/collaborativeagents/agents/local_user_agent.py +++ b/collaborativeagents/agents/local_user_agent.py @@ -11,9 +11,11 @@ from typing import List, Dict, Any, Optional from copy import deepcopy from json_repair import repair_json -# Default model paths -DEFAULT_MODEL_PATH_8B = "/projects/bfqt/users/yurenh2/ml-projects/personalization-user-model/models/llama-3.1-8b-instruct" -DEFAULT_MODEL_PATH_70B = "hugging-quants/Meta-Llama-3.1-70B-Instruct-AWQ-INT4" +# Default model paths - computed relative to project root +from pathlib import Path +_PROJECT_ROOT = Path(__file__).parent.parent.parent +DEFAULT_MODEL_PATH_8B = str(_PROJECT_ROOT / "models/llama-3.1-8b-instruct") +DEFAULT_MODEL_PATH_70B = str(_PROJECT_ROOT / "models/llama-3.1-70b-instruct") # Use 70B by default for better user simulation DEFAULT_MODEL_PATH = DEFAULT_MODEL_PATH_70B |