-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathrequirements.txt
More file actions
28 lines (21 loc) · 1.02 KB
/
requirements.txt
File metadata and controls
28 lines (21 loc) · 1.02 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
# Bootstrap Basil Dependencies
# Install with: pip install -r requirements.txt
# PyTorch - Deep learning framework for model training and inference
# Note: For GPU support, you may need to install a CUDA-specific version
# See: https://pytorch.org/get-started/locally/
torch>=2.0.0
# Hugging Face Transformers - Model loading and text generation
transformers>=4.30.0
# Hugging Face PEFT - Parameter-Efficient Fine-Tuning (LoRA adapters)
# Used for dual-objective training: Basil-policy LoRA adapters on GPT-2
peft>=0.13.0
# Hugging Face Accelerate - Required by PEFT for model handling
accelerate>=0.34.0
# OpenAI Python SDK - For GPT-4o-mini API calls (Tutor, Sophie, Grader, etc.)
openai>=1.0.0
# pyspellchecker - English dictionary for programmatic Englishness scoring
# Used by score_override.py to detect real English words in Basil's output
pyspellchecker>=0.8.0
# filelock - Cross-process file locking for parallel session generation
# Protects shared JSON state files (metrics, assessment, used_lessons, etc.)
filelock>=3.12.0