This repository contains the code for a Project Manager Co-Pilot Chatbot using a Hierarchical Joint Embedding Predictive Architecture (H-JEPA) with the Ollama model. The chatbot is trained using self-supervised learning and fine-tuned with task-specific data.
Inspired by the "A Path Towards Autonomous Machine Intelligence" Paper. Link to paper: https://openreview.net/forum?id=BZ5a1r-kVsf
-
Clone the repository: (Command Prompt - Run as Administrator on Windows OS)
git clone https://github.com/ai-in-pm/H-JEPA_PM_Co-Pilot.git
-
CD into Directory:
cd/H-JEPA_PM_Co-Pilot
-
Install the required dependencies:
pip install -r requirements.txt
-
Prepare your training data and place it in the
data/directory. -
Train the model:
python src/train.py
- Use the trained model for chatbot inference:
from src.model import generate_response
context = "How can I effectively manage project timelines?"
response = generate_response(context)
print(response)