paladim-1b-medical / requirements.txt
nickagge's picture
Upload requirements.txt with huggingface_hub
d59deca verified
Raw
History Blame Contribute Delete
867 Bytes
# PALADIM Dependencies
# Pre Adaptive Learning Architecture of Dual-Process Hebbian-MoE Schema
# Core
torch>=2.0.0
transformers>=4.30.0
# LoRA/Plastic Memory
peft>=0.5.0
# Distributed training
accelerate>=0.20.0
# Evaluation
scikit-learn>=1.2.0
# Utilities
numpy>=1.24.0
tqdm>=4.65.0
# Experiment tracking (optional)
wandb>=0.15.0
# Additional utilities
einops>=0.6.0
scipy>=1.10.0
# Web Control Panel & API
flask>=3.0.0
flask-cors>=6.0.0
# ONNX Export (for cross-platform deployment)
onnx>=1.12.0
onnxruntime>=1.13.0
onnxscript>=0.1.0
# Optional: For advanced optimization
# onnxruntime-tools>=1.7.0
# FSDP Training (for huge models across multiple GPUs)
# Requires PyTorch 2.0+ with CUDA
# Install: pip install torch>=2.0.0 --index-url https://download.pytorch.org/whl/cu118
# Dataset loading
datasets>=2.14.0