File size: 1,709 Bytes
4f1072f | 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 29 30 31 32 33 34 35 36 37 38 39 40 41 42 | # OncoAgent - Pinned Dependencies for Reproducibility # Hardware: AMD Instinct MI300X | ROCm 7.2.x # ================================================ # === Core ML Framework === torch>=2.3.0 # ROCm maps device="cuda" to HIP internally transformers>=4.42.0 # Llama 3.1 support accelerate>=0.31.0 # Distributed training utilities # === Fine-Tuning (QLoRA) === peft>=0.11.0 # LoRA/QLoRA adapters bitsandbytes>=0.43.0 # 4-bit quantization (ROCm-compatible build required) datasets>=2.20.0 # HuggingFace datasets loading trl>=0.9.0 # SFT Trainer # === RAG Engine === PyMuPDF>=1.24.0 # PDF extraction (fitz) pymupdf4llm>=0.0.17 # SOTA: PDF to Markdown with table support chromadb>=0.5.0 # Local vector database sentence-transformers>=3.0.0 # BioBERT embeddings + CrossEncoder re-ranking networkx>=3.0 # SOTA: Knowledge Graph / GraphRAG requests>=2.31.0 # SOTA: API connectivity (CIViC/ClinicalTrials) # === Agent Orchestration === langgraph>=0.1.0 # Multi-agent state graph — Rule #17 langchain-core>=0.2.0 # Base abstractions for agents # === Inference Engine === vllm>=0.5.0 # PagedAttention serving — Rule #6 # === UI === gradio>=4.36.0 # Frontend interface — Rule #29 # === Utilities === python-dotenv>=1.0.0 # .env secret management — Rule #24 huggingface-hub>=0.23.0 # Model/dataset downloads openai>=1.30.0 # Featherless.ai API (OpenAI-compatible) # === Testing & Dev === pytest>=8.0.0 # Test framework reportlab>=4.0.0 # Synthetic PDF generation for testing |